HDU 5790 Prefix 字符串哈希+主席树

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=5790

题意

给定n个字符串，然后m次询问，每次询问给出l,r代表在第l和第r个串之间本质不同的前缀有多少个

分析

本题是要求强制在线的，因此只能考虑主席树维护历史版本，但这题与普通主席树有些不同，需要自己变形一下。我们记录一下每个字符串的前缀对当前状态的贡献，例如当前前缀在之前版本出现过，我们加上当前版本的贡献，再减去上个出现过版本的贡献。最后求[l,r]区间实质就是求r版本[l,r]上的和。

至于怎么算前缀出现位置，可以用字符串哈希处理出来然后记录一下最后出现位置就可以了。

Code

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define re register
typedef long long ll;
typedef pair<ll, ll> PII;
typedef unsigned long long ull;
const int N = 1e5 + 10, M = 1e6 + 5, INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
struct node {
    int lc, rc;
    int sum;
}hjt[N << 5];
int rt[N], cnt;
void modify(int &now, int pre, int l, int r, int pos, int val) {
    now = ++cnt;
    hjt[now] = hjt[pre];
    hjt[now].sum += val;
    if (l == r) return;
    int mid = (l + r) >> 1;
    if (pos <= mid) modify(hjt[now].lc, hjt[pre].lc, l, mid, pos, val);
    else modify(hjt[now].rc, hjt[pre].rc, mid+1, r, pos, val);
}
int query(int now, int l, int r, int ql, int qr) {
    if (ql <= l && qr >= r) return hjt[now].sum;
    int mid = (l + r) >> 1;
    int ans = 0;
    if (ql <= mid) ans += query(hjt[now].lc, l, mid, ql, qr);
    if (qr > mid) ans += query(hjt[now].rc, mid+1, r, ql, qr);
    return ans;
}
vector<ull> hs[N];
map<ull, int> mp;
string s[N];
void init() {
    cnt = 0;
    mp.clear();
}
void solve() {
    int n; while (cin >> n) {
        init();
        for (int i = 1; i <= n; i++) {
            cin >> s[i];
            hs[i].resize(s[i].size());
            hs[i][0] = s[i][0];
            for (int j = 1; j < s[i].size(); j++) hs[i][j] = hs[i][j-1] * 23333 + s[i][j];
        }
        for (int i = 1, pre; i <= n; i++) {
            rt[i] = rt[i-1];
            for (auto &it : hs[i]) {
                pre = mp[it];
                modify(rt[i], rt[i], 1, n, i, 1);
                if (pre) modify(rt[i], rt[i], 1, n, pre, -1);
                mp[it] = i;
            }
        }
        int ans = 0;
        int m; cin >> m; while (m--) {
            int l, r; cin >> l >> r;
            l = (l + ans) % n + 1;
            r = (r + ans) % n + 1;
            if (l > r) swap(l, r);
            printf("%d\n", ans = query(rt[r], 1, n, l, r));
        }
    }
}


signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
#ifdef ACM_LOCAL
    freopen("input", "r", stdin);
    freopen("output", "w", stdout);
#endif
    solve();
    return 0;
}