841 字符串哈希

1. 问题描述： 

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

输入格式

第一行包含整数 n 和 m，表示字符串长度和询问次数。
第二行包含一个长度为 n 的字符串，字符串中只包含大小写英文字母和数字。
接下来 m 行，每行包含四个整数 l1，r1，l2，r2，表示一次询问所涉及的两个区间。
注意，字符串的位置从 1 开始编号。

输出格式

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

数据范围

1 ≤ n，m ≤ 10 ^ 5

输入样例：

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

输出样例：

Yes
No
Yes

2. 思路分析：

字符串哈希算法用来快速判断两段字符串是否相同，我们实际上是将字符串看成是一个P进制的数，其中P一般为131或13331这样产生冲突的概率会很小，可以理解为几乎没有冲突；一开始的时候使用O(n)的时间预处理一个字符串所有前缀的哈希值，然后我们就可以使用O(1)的时间查询出字符串的哈希值，这样就可以快速判断两个字符串是否相等，字符串哈希一般涉及到下图中的公式，包括计算一个字符串的哈希值，计算字符串中子串的哈希值等。下图是计算一个字符串的P进制数字的相关计算公式：

3. 代码如下：

java：

import java.util.Scanner;

public class Main{
    static long []h;
    static int P = 13331;
    static long []Q;
    static int n;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int m = sc.nextInt();
        // 接收输入两个数字之后的回车符
        sc.nextLine();
        String s = sc.nextLine();
        h = new long[n + 1];
        // Q用来存储p的n次幂
        Q = new long[n + 1];
        Q[0] = 1;
        hash(s);
        for (int i = 0; i < m; ++i){
            int l1 = sc.nextInt();
            int r1 = sc.nextInt();
            int l2 = sc.nextInt();
            int r2 = sc.nextInt();
            if (query(l1, r1) == query(l2, r2)) System.out.println("Yes");
            else System.out.println("No");
        }
    }
    
    // 预处理字符串所有前缀的哈希值, 将结果存储到h中
    public static void hash(String s){
        for (int i = 0; i < n; ++i){
            h[i + 1] = h[i] * P + (int)s.charAt(i);
            Q[i + 1] = Q[i] * P;
        }
    }
    
    // 直接使用公式计算出当前字符串对应的区间的哈希值
    public static long query(int l, int r){
        return h[r] - h[l - 1] * Q[r - l + 1];
    }
}

python代码（超时）：

from typing import List


# 计算字符串的哈希值与p的n次幂
def hash(h: List[int], s: str, P: int, Q: List[int]):
    for i in range(1, len(s) + 1):
        c = s[i - 1]
        # 使用字符串ascii码
        h[i] = h[i - 1] * P + ord(c)
        Q[i] = Q[i - 1] * P


# 使用公式计算即可
def query(h: List[int], l: int, r: int, Q: List[int]):
    return h[r] - h[l - 1] * Q[r - l + 1]


if __name__ == '__main__':
    n, m = map(int, input().split())
    s = input()
    h = [0] * (n + 1)
    P = 13331
    # Q存储P的i次幂, 先预处理之后直接查询得到
    Q = [0] * (n + 1)
    Q[0] = 1
    hash(h, s, P, Q)
    # h存储字符串的哈希结果
    for i in range(m):
        l1, r1, l2, r2 = map(int, input().split())
        if query(h, l1, r1, Q) == query(h, l2, r2, Q):
            print("Yes")
        else:
            print("No")