AC‘s String(字符串哈希线段树 + set + 乘法逆元)

题目链接: AC’s String

大致题意

给定n个字符串(作为模式串), 和一个匹配串. 有如下两种操作:

​ ①询问匹配串[l, r]区间的子串是否是某个模式串.
​ ②修改匹配串x位置, 使其位置的字符变为y.

解题思路

字符串哈希线段树 下面的思路不适合不了解字符串哈希方式, 以及不会线段树的同学食用

如何判断两个字符串是否相等? 判断其哈希值是否相等即可.

那么对于每一个匹配串的子串是否为某个模式串, 我们可以将所有模式串的哈希值放到一个集合set中, 每次判断当前子串的哈希值是否存在于集合中即可.

此时有一个问题, 如果是询问一个串的两个子串: [l1, r1] 与 [l2, r2]两部分是否相同. 那么当我们求完两个子串的哈希值后, 需要考虑到对齐问题. 我们只需要让前面的那个子串向后对齐哈希即可.

但是这个题, 你会发现是无法让前面的向后对齐的, 因此这里需要用到乘法逆元. 而最简单的逆元就是运用费马小定理+快速幂解决了. 因此我们在设置hash取模数的时候, 需要设定一个质数.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E5 + 10, B = 13331, mod = 1E9 + 7; //B为哈希基数
int P[N]; 

char s[N];
unordered_set<int> st;


char w[N];
struct node {
    int l, r; int base;
    int hash;
}t[N << 2];
void pushup(int x) { t[x].hash = (t[x << 1].hash + t[x << 1 | 1].hash) % mod; }

void build(int l, int r, int x = 1) {
    t[x] = { l, r, P[l], (int)(1ll * w[l] * P[l] % mod) };
    if (l == r) return;
    int mid = l + r >> 1;
    build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
    t[x].base = (t[x << 1].base + t[x << 1 | 1].base) % mod;
    pushup(x);
}

void modify(int a, char c, int x = 1) {
    if (t[x].l == t[x].r && t[x].l == a) {
        t[x].hash = 1ll * c * t[x].base % mod;
        return;
    }
    int mid = t[x].l + t[x].r >> 1;
    modify(a, c, x << 1 | (a > mid));
    pushup(x);
}

int ask(int l, int r, int x = 1) {
    if (l <= t[x].l && r >= t[x].r) return t[x].hash;
    int mid = t[x].l + t[x].r >> 1;
    int res = 0;
    if (l <= mid) res = ask(l, r, x << 1);
    if (r > mid) res += ask(l, r, x << 1 | 1);
    return res % mod;
}

void init() {
    P[0] = 1;
    rep(i, N - 5) P[i] = (ll)P[i - 1] * B % mod;
}

int gethash(char* s, int len) { //注意要采用与线段树相同的哈希方式
    int res = 0;
    for (int i = 0; i < len; ++i) {
        res = (res + 1ll * s[i] * P[i + 1]) % mod;
    }
    return res;
}

int fpow(int a, int b, int p) { 
    ll res = 1; a %= p;
    while (b) {
        if (b & 1) res = res * a % p;
        b >>= 1;
        a = 1ll * a * a % p;
    }
    return res;
}
int main()
{
    init();
    int T; cin >> T;
    rep(tt, T) {
        printf("Case #%d:\n", tt);
        
        st.clear();
        int n; scanf("%d", &n);
        rep(i, n) {
            scanf("%s", s);
            st.insert(gethash(s, strlen(s)));
        }
        
        memset(t, 0, sizeof t);
        scanf("%s", w + 1);
        build(1, strlen(w + 1));
        
        int m; scanf("%d", &m);
        while (m--) {
            char op[2]; scanf("%s", op);
            if (*op == 'Q') {
                int l, r; scanf("%d %d", &l, &r);
                int res = ask(l + 1, r + 1);
                res = 1ll * res * fpow(P[l], mod - 2, mod) % mod;
                printf("%s\n", st.count(res) ? "Yes" : "No");
            }
            else {
                int a; char c[2]; scanf("%d %s", &a, c);
                modify(a + 1, c[0]);
            }
        }
    }
    return 0;
}

END