【HPU 1415: 小ho的01串】& 尺取

1415: 小ho的01串 [字符串]

时间限制: 1 Sec 内存限制: 128 MB

提交: 269 解决: 131 统计

题目描述

有一个由0和1组成的字符串，它好长呀——–一望无际

恩，说正题，小ho的数学不太好，虽然是学计算机的但是看见0和1也是很头疼的，

现在他的老师想让他计算出来包含K个1的子串有多少个—–呀，头要炸了！！！

小ho知道你的数学棒棒哒，所以来找你帮忙了。

输入

第一行是一个数K。

第二行是一个字符串str。

0 < |str| ≤ 106

输出

一个数S,代表了满足条件的个数。

样例输入

2

101010

样例输出

6

来源

Leibniz_Zhang

用数组记录下 1 的下标，维护区间 [l,r] 中 1 的个数为 k,这个区间的贡献等于 （a[l] - a[l - 1] ）* （a[r + 1] - a[r]），会出现 010100 的情况在末尾加个 1 即可，k == 0, 相当于求 1 与 1 之间 0 的组合

AC代码：

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX = 1e6 + 10;
typedef long long ll;
int k,a[MAX];
char s[MAX];
void solve(int o){
ll l = 1,r = 0,ans = 0,sum = 0;
a[0] = -1;
for(int i = 0; i <= o; i++)
if(s[i] == '1'){
a[++r] = i,sum++;
if(sum == k + 1) ans += (a[l] - a[l - 1]) * (a[r] - a[r - 1]),l++,sum--;
}
printf("%lld\n",ans);
}
int main()
{
while(~scanf("%d %s",&k,s)){
ll o = strlen(s),l = 0,r = 0,ans = 0;
s[o] = '1';
if(!k){
for(int i = 0; i <= o; i++)
if(s[i] == '1')
r = i - l,l = i + 1,ans += r * (r + 1) / 2;
printf("%lld\n",ans + (r + 1) * r / 2);
}
else solve(o);
}
return 0;
}