小ho的01串 [字符串]

#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;
}