AcWing 841. 字符串哈希

知识点：根据哈希值，解决字符串匹配问题;非常牛逼的一个做法，比kmp更简单

给定一个长度为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

算法思想：

如果[l1,r1]的哈希值与[l2,r2]的哈希值一样，则表示区间内的字符串相等！

h[r]表示[0,r]的字符串的哈希值

h[l]表示[o,l]的字符串的哈希值

我们要计算出[l,r]区间的哈希值，因此我们要计算出所有前缀字符串的哈希值

例如aabcde，我们要计算 a、aa、aab、aabc、aabcd、aabcde的哈希值

哈希值计算方式为

h[k] = str[0] * p^(k - 1) + str[1] *p^(k - 2) + ... + str[k - 1] * p^0

则

h[r] = str[0] * p^(r - 1) + str[1] *p^(r - 2) + ... + str[r - 1] * p^0

h[l - 1] = str[0] * p^(l - 2) + str[1] *p^(l - 3) + ... + str[l - 1] * p^0

他们之间不能直接相减，因为他们的层级不一样，需要将l位置的层级向左移动(r - 1) - (l - 2) = r - l + 1保证对对齐

则[l,r]的哈希值为

h[r] - h[l - 1] * p^(r - l + 1)

import java.io.*;
import java.lang.*;

class Main{
    
    static int N = 100010, P = 131;//进制
    static long[] h = new long[N];//存储字符串哈希值
    static long[] p = new long[N];//存储当前位置的指数层级
    static long getLR(int l, int r){
        return h[r] - h[l - 1] * p[r - l + 1];
    }
    
    public static void main(String[] args)throws Exception{
        BufferedReader buf = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter buw = new BufferedWriter(new OutputStreamWriter(System.out));
        String[] params = buf.readLine().split(" ");
        int n = Integer.valueOf(params[0]);
        int m = Integer.valueOf(params[1]);
        String str = buf.readLine();
        p[0] = 1;
        for(int i = 1; i <= n; ++i){//计算指数层级和字符串哈希值
            p[i] = p[i - 1] * P;
            h[i] = h[i - 1] * P + str.charAt(i - 1) - 'a';
        }
        while(m -- != 0){
            String[] info = buf.readLine().split(" ");
            int l1 = Integer.valueOf(info[0]);
            int r1 = Integer.valueOf(info[1]);
            int l2 = Integer.valueOf(info[2]);
            int r2 = Integer.valueOf(info[3]);
            if(getLR(l1, r1) == getLR(l2, r2)){
                buw.write("Yes\n");
            }else{
                buw.write("No\n");
            }
        }
        buw.flush();
        buf.close();
        buw.close();
        
    }
}