算法题 手写实现字符串哈希

题目

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

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

输入格式

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

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

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

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

输出格式

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

每个结果占一行。

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：

8 3
aabbaabb
1 3 5 7
1 3 6 8
1 2 1 2

输出样例：

Yes
No
Yes

C++实现

#include <iostream>
using namespace std;

typedef unsigned long long ULL;

const int N = 1e5+10, P = 131; # 根据经验得到，不容易冲突的质数

int n, m;
int h[N], p[N];
char str[N];

ULL get(int l, int r){
    return h[r]-h[l-1]*p[r-l+1]; # 获取区间内的hash值
}

int main(){
    scanf("%d%d%s", &n, &m, str+1);
    p[0] = 1;
    h[0] = 1;
    for (int i = 1; i <= n; i++){
        p[i] = p[i-1] * P;
        h[i] = h[i-1] * P + str[i];
    }
    
    while(m--){
        int ll, lr, rl, rr;
        scanf("%d%d%d%d", &ll, &lr, &rl, &rr);
        //printf("%d %d", get(ll, lr), get(rl, rr));
        if (get(ll, lr) == get(rl, rr)) puts("Yes");
        else puts("No");
    }
    return 0;
}

Python实现

n, m = map(int, input().split())
s = '0'+input()
P = 131
p = [1]*(n+1)
h = [1]*(n+1)
mod = int(2**64)
for i in range(1, n+1):
    p[i] = p[i-1] * P % mod
    h[i] = (h[i-1] * P + ord(s[i])) % mod

def get(l, r):
    return (h[r] - h[l-1] * p[r-l+1]) % mod
    
for _ in range(m):
    ll, lr, rl, rr = map(int, input().split())
    if get(ll, lr) == get(rl, rr): print('Yes')
    else: print('No')