2018 美团校招笔试题 字符串距离

题目描述

给出两个相同长度的由字符 a 和 b 构成的字符串，定义它们的距离为对应位置不同的字符 的数量。如串”aab”与串”aba”的距离为 2;串”ba”与串”aa”的距离为 1;串”baa”和串”baa”的 距离为 0。下面给出两个字符串 S 与 T，其中 S 的长度不小于 T 的长度。我们用|S|代表 S 的 长度，|T|代表 T 的长度，那么在 S 中一共有|S|-|T|+1 个与 T 长度相同的子串，现在你需要计 算 T 串与这些|S|-|T|+1 个子串的距离的和。

输入描述:

第一行包含一个字符串 S。

第二行包含一个字符串 T。
S和T均由字符a和b组成，1≤ |T| ≤ |S| ≤105。

输出描述:

输出对应的答案。

示例1

输入

aab
aba

输出

2

示例2

输入

aaabb
bab

输出

5

说明

在样例 2 中，”aaa”与”bab”的距离为 2，”aab”与”bab”的距离为 1，”abb”与”bab”的距离为 2，
所以最后答案为 5。

这道题目直接用滑窗比较的方法，时间复杂度为O(N^2) , 可以通过70%

存在O(N) 的做法是注意到, 对于每一个t[i]字符，与字符串s[i]到s[i+k]均比较了一次，其中k=len(s)-len(t)， 所以想到用前缀和优化

/*
两个循环会超时, 通过700%
扫描思想：例如s=aaabb，t=bab
1.先看t中的第一个字符b在s中扫过s=[0:2]=aaa 所以有三次不同
2.再看t中的第二个字符a在s中扫过s=[1:3]=aab 所以有一次不同
3.再看t中的第三个字符b在s中扫过s=[2:4]=abb 所以有一次不同
建立a[i]代表s中s[:i]中a的个数，那么b的个数就是i-s[:i]
假设len(s)-len(t)=k
4. 如果t[i] = 'a', 那么t[i] 扫过 s[i:i+k], 这个区间之和是 a[i+k]-a[i]

*/

#include<iostream>
#include<string>
 
using namespace std;
const int N = 2000100;
typedef long long LL;
int Hash[N]; // 表示前i个字母中a出现的次数 i从1开始，代表第一个字母

int main(){
    string s,t;
    cin >> s >> t;
    for(int i=1;i<=s.size();i++){
        if(s[i-1]=='a'){
            Hash[i]=Hash[i-1]+1;
        }
        else
            Hash[i]=Hash[i-1];
    }
    int k = s.size()-t.size();
    LL res = 0;
    for(int i=0;i<t.size();i++){
        if(t[i]=='b') res+= (Hash[i+k+1]-Hash[i]);
        else res+= (i+k+1-Hash[i+k+1] - (i-Hash[i])); 
    }
    cout<<res<<endl;
    
}