cf#336-B-Hamming Distance Sum 前缀和

传送门：cf#336-B-Hamming Distance Sum

题目大意

给定两个字符串a,b，求b串中长度为a的子串中的Hamming Distance之和

解题思路

根据数据范围所以肯定不能有两层循环。
我们可以先求出b串从第一个到某一位中字符1的个数，one[i]就表示b前i个位置中字符1的个数。
然后我们在挨个遍历a字符串，如果a[i]能够访问的b重最长的下标为lenB-lenA+i，所以如果a[i]等于字符0的时候sum += one[能访问到的最远下标]-one[访问的第一个下标]，当a[i]等于字符1的时候也就变为上面的相反情况只需要用能访问到总长度-上面的额情况就可以了，也就是sum+= lenB-lenA+1-one[能访问到的最远下标]-one[访问的第一个下标];

AC代码

#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN = 200008;
int main()
{
    char a[MAXN];
    char b[MAXN];
    while(~scanf("%s%s",a+1,b+1))
    {
        int lenA = strlen(a+1);
        int lenB = strlen(b+1);
        int one[MAXN];
        memset(one,0,sizeof one);
        for(int i=1;i<=lenB;i++)
            if(b[i] == '1')
                one[i] = one[i-1]+1;
            else
                one[i] = one[i-1];
        int sum = 0;
        for(int i=1;i<=lenA;i++)
        {
            if(a[i] == '1')
            {
                sum+=lenB-lenA+1-(one[lenB-lenA+i]-one[i-1]);
            }
            else
            {
                sum+=(one[lenB-lenA+i]-one[i-1]);
            }
        }
        printf("%d",sum);
    }
    return 0;
}