HPU1415--小ho的01串

题目描述

有一个由

0

和

1

组成的字符串，它好长呀--------一望无际

恩，说正题，小ho的数学不太好，虽然是学计算机的但是看见

0

和

1

也是很头疼的，

现在他的老师想让他计算出来包含

K

个

1

的子串有多少个-----呀，头要炸了！！！

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

输入

第一行是一个数

K

。

第二行是一个字符串

str

。

0 < |str| ≤ 106

输出

一个数

S

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

样例输入

2
101010

样例输出

6

求包含k个1的子串有多少个

枚举子串中第一个1的位置x，然后找到它后面的第k个1的位置y

对于这个情况子串就有n*m个(n为x前面连续0的个数+1，m为y后面连续0的个数+1)

#include <cstdio>
#include <cstring>
int zero[1000005];
char st[1000005];
int main()
{
int k;
scanf("%d",&k);
scanf("%s",st);
int len = strlen(st);
int cz = 0;
for (int i = 0;i < len; ++i)
{
if (st[i]=='1') cz++;
else zero[cz]++;//记录每个位置的1后面0的个数
}
cz = 0;
int ans = 0;
int cnt = 0;
for (int i = 0;i < len; ++i)
{
if (st[i]=='1')
{
cnt++; cz++;
}
if (cnt == k)
{
ans += (zero[cz-k]+1)*(zero[cz]+1); //j-k前1个1后面0的个数*j后0的个数(+1是因为还有不算0的情况)
cnt--;
}
}
printf("%d\n",ans);
return 0;
}

第二种方法：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<list>
#include<cmath>
#include<vector>
using namespace std;
const int maxn=1000010;
char str[maxn];
long long num[maxn];
int main()
{
long long ans=0,k,cnt=1;
scanf("%lld%s",&k,str);
long long l=1,r,len=strlen(str),temp=0;num[0]=-1;
if(k==0)
{
l=0;
for(long long i=0;i<len;++i)
{
if(str[i]=='1')
{
r=i-l;l=i+1;
ans=ans+r*(r+1)/2;
}
}
r=(len-l);
ans=ans+r*(r+1)/2;
printf("%d\n",ans);
return 0;
}
for(long long i=0;i<len;++i)
{
if(str[i]=='1')
{
temp++;num[cnt++]=i;
}
if(temp==k+1)
{
ans=ans+(num[l]-num[l-1])*(num[cnt-1]-num[cnt-2]);l++;temp--;
}
}
if(temp==k)
{
ans=ans+(num[l]-num[l-1])*(len-num[cnt-1]);
}
printf("%d\n",ans);
return 0;
}