Another Problem on Strings CodeForces - 165C

原题链接

考察:前缀和+二分+双指针

补一个大佬的神级解法 Go

错误思路:

      纯双指针处理,写了几个小时没写出来.....

思路:

       前缀和预处理,sum[i]表示1~i坐标有多少个1,那么[l,r]区间内1的个数为sum[r] - sum[l-1],那么我们可以枚举端点,计算[l,r]区间里有多少个1,如果==k,ans++.

       以上为暴力思路,根据数据我们还需要优化.一般优化第二层循环.枚举左端点,找到==sum[i]+k的sum[j]和>sum[i]+k的sum[r],[j,r)之间都为答案.

       但是要特判k = 0,这里本蒟蒻用的双指针处理.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 1000010;
int k,sum[N];
char s[N];
LL ans;
int main()
{
    scanf("%d%s",&k,s+1);
    int len = strlen(s+1),cnt = 0;
    for(int i=1;i<=len;i++) sum[i] = sum[i-1]+(s[i]-'0');
    if(!k)
    {
        for(int i=1,j=1;i<=len;i++)
        {
            int t = s[i]-'0';
            if(t) continue;
            if(j<i) j = i;
            while(j<=len&&s[j]=='0') j++;
            ans+=(LL)(j-i);
        }
        printf("%I64d\n",ans);
        return 0; 
    }
    for(int i=0;i<len;i++)
    {//sum[l~r]-sum[i] = k
        int l = lower_bound(sum+1,sum+len+1,sum[i]+k)-sum;
        int r = upper_bound(sum+1,sum+len+1,sum[i]+k)-sum;
        if(l==len+1) break;//当l==len+1,r一定==len+1,r-l==0; 
        ans+=(LL)(r-l);
    }
    printf("%I64d\n",ans);
    return 0;
}