题意是:有字符串a和b(由0和1组成),a的长度小于等于b,求b中每个和a长度相同的子串的差值和,差值计算相同为0,不同为1,比如0011和1101的差值是3。
赛时没有想到前缀和,这题有前缀和就可以解决了。
1 #include <bits/stdc++.h> 2 #define ll long long 3 using namespace std; 4 const int MAX = 2e5+10; 5 char s[MAX], ss[MAX]; 6 int a[MAX]; 7 int main(){ 8 scanf("%s",s); 9 scanf("%s",ss); 10 int len = strlen(s), len2 = strlen(ss); 11 for(int i = 0; i < len2; i ++){ 12 if(ss[i] == '1') 13 a[i+1] ++; 14 a[i+1] += a[i]; 15 } 16 ll ans = 0; 17 int k = len2-len+1; 18 for(int i = 0; i < len; i ++){ 19 if(s[i] == '1') ans += k-a[k+i]+a[i]; 20 else ans += a[k+i]-a[i]; 21 } 22 cout << ans << endl; 23 return 0; 24 }