【算法】字符串哈希

问题描述

给定一个长度为 n 的字符串，再给定 m个询问，每个询问包含四个整数 l1,r1,l2,r2，请你判断 [l1,r1] 和 [l2,r2] 这两个区间所包含的字符串子串是否完全相同。

字符串中只包含大小写英文字母和数字。

输入格式
第一行包含整数 n 和 m，表示字符串长度和询问次数。

第二行包含一个长度为 n 的字符串，字符串中只包含大小写英文字母和数字。

接下来 m 行，每行包含四个整数 l1,r1,l2,r2，表示一次询问所涉及的两个区间。

注意，字符串的位置从 1 开始编号。

输出格式
对于每个询问输出一个结果，如果两个字符串子串完全相同则输出 Yes，否则输出 No。

每个结果占一行。

问题分析

我们尝试对字符串进行编码
h[i]为从第一个字符到第i个字符编码得到的数字
计算方法如下图：

求第l个字符到第r个字符编码得到的数字：

完整代码

# include <iostream>
# include <cstring>
# include <cmath>

using namespace std;
const int p = 131;
const int N =1e5+5;
unsigned long long h[N],pp[N];



unsigned long long query(int l,int r){
     return h[r] - h[l-1]*pp[r-l+1];
 }

int main(){
    int n,m;
    cin>>n>>m;
    string str;
    cin>>str;
    int i;
    h[0] = 0;
    pp[0] = 1;
    for(i=0; i<n;i++){
        pp[i+1] = pp[i] * p;
        h[i+1] = h[i] * p + str[i];
        //cout<<h[i+1]<<endl;
    }
    
    while(m--){
        int l1,r1,l2,r2;
        cin>>l1>>r1>>l2>>r2;
        if(query(l1,r1) == query(l2,r2)){
            cout<<"Yes"<<endl;
        }
        else{
            cout<<"No"<<endl;
        }
        
    }
    
}