题意:给两个01串S和T,求两个串的距离,即与S等长度的T的子串与S不同的位数(说起来有点绕,看样例好理解)
数据范围: |S|, |T| <= 2e5
思路:由数据范围可知不能瞎暴力,容易发现S的第一位可以与T的第一位到第lenT-lenS+1位对应,第i位可以与第lenT-lenS+i位对应,只要做一遍前缀和,逐位求贡献就行了,前缀和还是很骚气好用的技能orz
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #define LL long long 5 using namespace std; 6 7 const int mx = 2e5+10; 8 char a[mx], b[mx]; 9 int sum[mx]; 10 11 int main(){ 12 LL ans = 0; 13 scanf("%s%s", a+1, b+1); 14 int n = strlen(a+1), m = strlen(b+1); 15 for (int i = 1; i <= m; i++){ 16 if (b[i] == '0') sum[i] = sum[i-1]+1; 17 else sum[i] = sum[i-1]; 18 } 19 for (int i = 1; i <= n; i++){ 20 if (a[i] == '1') ans += sum[m-(n-i)]-sum[i-1]; 21 else ans += (m-(n-i)-sum[m-(n-i)])-(i-1-sum[i-1]); 22 } 23 printf("%lld ", ans); 24 return 0; 25 }