算法竞赛进阶指南 兔子与兔子

138. 兔子与兔子

很久很久以前，森林里住着一群兔子。

有一天，兔子们想要研究自己的 DNA 序列。

我们首先选取一个好长好长的 DNA 序列（小兔子是外星生物，DNA 序列可能包含 26 个小写英文字母）。

然后我们每次选择两个区间，询问如果用两个区间里的 DNA 序列分别生产出来两只兔子，这两个兔子是否一模一样。

注意两个兔子一模一样只可能是他们的 DNA 序列一模一样。

输入格式
第一行输入一个 DNA 字符串 S。

第二行一个数字 m，表示 m 次询问。

接下来 m 行，每行四个数字 l1,r1,l2,r2，分别表示此次询问的两个区间，注意字符串的位置从 1 开始编号。

输出格式
对于每次询问，输出一行表示结果。

如果两只兔子完全相同输出 Yes，否则输出 No（注意大小写）。

数据范围
1≤length(S),m≤1000000
输入样例：
aabbaabb
3
1 3 5 7
1 3 6 8
1 2 1 2
输出样例：
Yes
No
Yes
思路分析：这里我们用字符串hash算法，设F[i]表示前缀子串S[1~i]的hash值，则有F[i]=F[i-1]*131+(s[i]-‘a’+1)，这里的131是我们所选的一个质数，于是可以得到任何一个区间[l,r]的hash值为F[r]-F[l-1]*131^r-l+1。当两个区间的hash值相同时，我们就认为两个子串完全相同，整个算法的时间复杂度为O（length（s）+q）

#include<iostream>
#include<cstring>
using namespace std;
const int N=1000010;
unsigned long long f[N],p[N];
char s[N];
int main(){
    scanf("%s",s+1);
    int n=strlen(s+1),q;
    cin>>q;
    p[0]=1;
    for(int i=1;i<=n;i++){
        f[i]=f[i-1]*131+(s[i]-'a'+1);
        p[i]=p[i-1]*131;//保存131的i次方
    }
    for(int i=1;i<=q;i++){
        int l1,l2,r1,r2;
        cin>>l1>>r1>>l2>>r2;
        if(f[r1]-f[l1-1]*p[r1-l1+1]==f[r2]-f[l2-1]*p[r2-l2+1])//判断两个区间的hash值是否相同
        cout<<"Yes"<<endl;
        else
        cout<<"No"<<endl;
    }
    return 0;
}