Try Again

Codefroces B. Hamming Distance Sum

Genos needs your help. He was asked to solve the following programming problem by Saitama:

The length of some string s is denoted |s|. The Hamming distance between two strings s and t of equal length is defined as , where si is the i-th character of s and ti is the i-th character of t. For example, the Hamming distance between string "0011" and string "0110" is |0 - 0| + |0 - 1| + |1 - 1| + |1 - 0| = 0 + 1 + 0 + 1 = 2.

Given two binary strings a and b, find the sum of the Hamming distances between a and all contiguous substrings of b of length |a|.

Input

The first line of the input contains binary string a (1 ≤ |a| ≤ 200 000).

The second line of the input contains binary string b (|a| ≤ |b| ≤ 200 000).

Both strings are guaranteed to consist of characters '0' and '1' only.

Output

Print a single integer — the sum of Hamming distances between a and all contiguous substrings of b of length |a|.

Examples
Input
01
00111
Output
3
Input
0011
0110
Output
2
Note

For the first sample case, there are four contiguous substrings of b of length |a|: "00", "01", "11", and "11". The distance between "01" and "00" is |0 - 0| + |1 - 0| = 1. The distance between "01" and "01" is |0 - 0| + |1 - 1| = 0. The distance between "01" and "11" is |0 - 1| + |1 - 1| = 1. Last distance counts twice, as there are two occurrences of string "11". The sum of these edit distances is 1 + 0 + 1 + 1 = 3.

The second sample case is described in the statement.

找规律,字符匹配

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
using namespace std;
typedef long long ll;
int main()
{
    string a,b;
    cin>>a>>b;
    ll sum=0,ans=0;
    int x=a.size(),y=b.size();
    int z=y-x+1;
    for(int i=0;i<z;i++)
    {
        if(b[i]=='1') sum++;
    }
    for(int i=0;i<x;i++)//匹配1
    {
        if(a[i]=='1') ans+=z-sum;
        else ans+=sum;
        if(b[i]=='1')sum--;
        if(b[i+z]=='1') sum++;
    }
    cout<<ans<<endl;
    return 0;
}

 

posted @ 2017-07-05 16:21  十年换你一句好久不见  阅读(432)  评论(0编辑  收藏  举报