每日一题：01序列

01序列 - 题目 - Daimayuan Online Judge

大致题意是字符串中只有0和1，找出有多少个子串含有k个1 

依次记录每个1的下标，枚举每个1，运用乘法原理，比如01010100，找含有2个1的子串，先枚举第一个橙色的1，然后如果找到第2个橙色的1，从第一个橙色的1开始往左数(包括它自己)，数尽所有的0(数到上一个1右边的0为止)，乘从第二个橙色的1开始往右数(包括它自己)，数尽所有的0(数到下一个1左边的0为止)，即2*2=4，再枚举第二个橙色的1，同理2*3=6，加起来就是10

然后交上去代码是错误的，原因是这只适用于k不为0的情况，如果k为0，数的时候就要避开1，于是分情况讨论，k不为0的时候就是上面那种情况，当k为0的时候，也用乘法原理，枚举每一个0，比如0001，含有第一个0的子串个数就是1*3，含有第二个0的子串个数就是0*2，含有第三个0的子串个数就是0*1，加起来就行了

乘法原理具体解释见(3条消息) 每日一题：子串的最大差_m0_74087709的博客-CSDN博客 

AC代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#define int long long
using namespace std;
const int N = 1e6 + 10;
int a[N];
int cnt;
signed main()
{
    int k;
    string s;
    cin >> k >> s;
    for (int i = 0; i < s.size(); i++) {
        if (s[i] == '1') {
            cnt++;
            a[cnt] = i;
        }
    }
    int res = 0;
    a[0] = -1;
    if (k == 0) {
        for (int i = 1; i <= cnt; i++) {
            for (int j = a[i - 1] + 1; j <= a[i] - 1; j++) {
                res += a[i] - j;
            }
        }
        for (int j = a[cnt] + 1; j < s.size(); j++) {
            res += s.size() - j;
        }

    }
    else
    for (int i = 1; i <= cnt; i++) {
        if (i + k-1 <= cnt) {
            if(i+k<=cnt)
            res += (a[i] - a[i - 1]) * (a[i + k] - a[i + k - 1]);
            else {
                res+= (a[i] - a[i - 1]) * (s.size() - a[i + k - 1]);
            }
        }
    }
    cout << res << endl;
}