CodeForces 608B Hamming Distance Sum

版权声明：本文为博主原创文章，转载请联系博主（通常都会同意就是了） https://blog.csdn.net/cccccwb/article/details/86608448

题解：像这种求一堆区间的某些值的和的，常常考虑成每个位置对最终答案的贡献的和，所以就考虑每个位置的贡献，稍加推算就能知道每个位置的a最终会对应哪一段的b，这个位置如果是1，那么做出的贡献就是对应区间的0的数量，反之亦然。所以只要先统计一下b的前缀和，然后扫一遍a就可以得出答案了。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

int la,lb;
long long ans;
int a[200010],b[200010],sb[200010];
char s[200010];

int main()
{
	gets(s);la=strlen(s);for(int i=1;i<=la;i++) a[i]=s[i-1]-48;
	gets(s);lb=strlen(s);for(int i=1;i<=lb;i++){b[i]=s[i-1]-48;sb[i]=sb[i-1]+b[i];}
	for(int i=1;i<=la;i++) 
	{
	   if(a[i]) ans+=lb-la+1-(sb[lb-la+i]-sb[i-1]);
	   else ans+=sb[lb-la+i]-sb[i-1];
    }
    cout<<ans<<endl;
	return 0;
}