codeforces 之 Another Problem on Strings

       赶脚自己逗比了，从昨天晚上开始刷这道题，前前后后交了12遍，最后实在是找不到错哪了，就是抱着试试看的态度改了一下数据类型，发现竟然是跪在了LONG LONG 这个整数类型上，这以后没法说自己是搞ACM的了。。。。。

      解题思路：

                给你一个只包含0和1的字符串，在这个字符串中你找一下包含k个1的子字符串的个数，所以当k=0时，或者在开头或者在任意两个1之间或者在结尾0的个数为co，经分析可知，0可以单个出现，可以最少两个的出现，可以找到规律：（co+1)*co/2; 当k > 0时，就要统计下在这k个1前面有a个0，后面有b个0，经分析可知：这k个1可以跟前面a个0一个或者多个0结合，同理可以跟后面的，也可以前后结合，可以找到规律：a+b+a*b。

 

代码如下：

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long a[1000010]={0};
char s[1000010];
int main()
{
 long long i,k,len,co=0;
 cin>>k;
 long long sum=0;
 cin>>s;len = strlen(s);
 for(i=0; s[i]; i++) {
  if(s[i] == '1') a[co++] = i;
 }
 if(k>co) cout<<0<<endl;
 else if(k==0) {
  if(co==0) {
   sum = (len+1)*len/2;
   cout<<sum<<endl;
   return 0;
  }
  sum=(a[0]*(a[0]+1))/2;
  for(i=1; i<co; i++) sum+=((a[i]-a[i-1]-1)*(a[i]-a[i-1]))/2;
  sum+=((len-a[co-1])*(len-a[co-1]-1))/2;
  cout<<sum<<endl;
 }
 else  {
  sum=co-k+1;
  for(i=k-1; i<co; i++) {
   long long b,c;
   if(i==k-1)  {
    b = a[0];
   }
   else {
    b = a[i-k+1] - a[i-k]-1;
   }
   if(i==co-1) {
    c = len-a[co-1]-1;
   }
   else {
    c = a[i+1] - a[i]-1;
   }
   sum+=(b+c+b*c);
  }
  cout<<sum<<endl;
 }
 return 0;
}