CF 165C dp

给定了一个01串，长度小于10^6

给了一个k值，问有多少个字串满足 这个字串中1的个数等于k。

预处理：

用map级记录下从左到有第一次到达cnt个1的下标。

用一个数组记录前i个字符中字符1的个数。

dp

然后，用dp[i] 表示s中以s[i]结尾的满足条件的字串的个数，

每次找到满足以i结尾的字串，开头有多少种情况，累加到ans里面。（具体看代码）

trick；我刚开始没有考虑k=1的情况，wa了一次。

#include <iostream>
#include <cmath>
#include <algorithm>
#include <string>
#include <deque>
#include <cstring>
#include <cstdio>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <cstdlib>
#include <iomanip>
using namespace std;

#define rep(i, a, b) for( i = (a); i <= (b); i++)
#define reps(i, a, b) for( i = (a); i < (b); i++)
#define pb push_back
#define ps push
#define mp make_pair
#define CLR(x,t) memset(x,t,sizeof x)
#define LEN(X) strlen(X)
#define F first
#define S second
#define Debug(x) cout<<#x<<"="<<x<<endl;


const double euler_r = 0.57721566490153286060651209;
const double pi = 3.141592653589793238462643383279;
const double E = 2.7182818284590452353602874713526;
const int inf=~0U>>1;
const int MOD = int(1e9) + 7;
const double EPS=1e-6;

typedef long long LL;
#define M 1000010
string s;
map<int, int> m;
LL ans;
int k, i, c[M];;
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    cin >> k >> s;
    int cnt, l = s.length();
    if(k == 0)
    {
        for(i = 0; i < l; i++)
        {
            cnt = 0;
            while(s[i] == '0') {cnt++; i++;}
            ans = ans + 1LL * cnt * (cnt - 1) / 2 + cnt;
        }
        cout << ans << endl;
        return 0;
    }
    cnt = 0;
    for(i = 0; i < l; i++)
    {
        if(s[i] == '1') {cnt++; m[cnt] = i;}
        c[i] = cnt;
    }
    for(i = 0; i < l; i++)
    {
        if(c[i] < k) continue;
        else if(c[i] == k)
        {
            int x = m[1];
            x++; //下标从0开始
            ans += x;

        }
        else
        {
            int d = c[i] - k;
            int be = m[d];
            int en = m[d + 1];
            int x = en - be;
            ans += x;
        }
        //cout << i <<  " " << ans << endl;
    }
    cout << ans << endl;
    return 0;
}