codeforces 336 Div.2 B. Hamming Distance Sum

题目链接：http://codeforces.com/problemset/problem/608/B

题目意思：给出两个字符串 a 和 b，然后在b中找出跟 a 一样长度的连续子串，每一位进行求相减的绝对值然后相加（这个讲得有点绕），直接举个例子就很容易理解了。

　　假设a = 01，b = 00111，

　　符合条件的b的子串有：00， 01， 11， 11

　　然后： |01-00| = |0-0| + |1-0| = 1， 

　　　　　 |01-01| = |0-0| + |1-1| = 0，

 　　　　　|01-11| = |0-1| + |1-1| = 1，

　　 　　　|01-11| = |0-1| + |1-1| = 1，

　　再求和：1 + 0 + 1 + 1 = 3

　　这个挺考观察能力的= =

　　枚举 a 的每一位，然后找出 b 中哪些子串需要跟a的这一位进行运算的。

　　假设 a = 010，b = 00111，la：a的字符串长度， lb：b的字符串长度

　　第 i 位　　　　　　需要 b 的 哪几位进行运算    

　　1　　　　　　　　 　1， 2， 3　   

　　2　　　　　　　　 　2， 3， 4

　　3　　　　　　　　 　3， 4， 5

　　然后有个这样的关系：对于字符串a中第 i 位，需要b中 i ~ lb-(la-i)   的这些数进行运算。  

　　可以发现，如果 a 中第 i 位的数为1， 那么b中 i ~ lb-(la-i)   的这些数为 0 跟它相减才等于1，否则为0； a 为 0 的话亦然。。。。最后就是注意用 long long 啦，10^18很大呢~

　　

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

const int maxn = 200000 + 5;
char a[maxn], b[maxn];
int c0[maxn], c1[maxn];

long long res;

int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt", "r", stdin);
    #endif // ONLINE_JUDGE

    while (scanf("%s%s", a+1, b+1) != EOF) {
        int la = strlen(a+1);
        int lb = strlen(b+1);

        memset(c0, 0, sizeof(c0));
        memset(c1, 0, sizeof(c1));

        for (int i = 1; i <= lb; i++) {
            c1[i] = c1[i-1];
            c0[i] = c0[i-1];

            if (b[i] == '0') {
                c0[i]++;
            }
            else {
                c1[i]++;
            }
        }

        res = 0;
        for (int i = 1; i <= la; i++) {
            if (a[i] == '0') {   // 统计跟它比对的1的个数
                res += (c1[lb-(la-i)]-c1[i-1]);
            }
            else {
                res += (c0[lb-(la-i)]-c0[i-1]);
            }
        }
        printf("%lld\n", res);
    }
    return 0;
}