[HDU5790] Prefix 字典树+主席树

Prefix

题意：

给出总长$10^5的N个串$，给出$q个询问$，查询编号$[l,r]$范围内的串有多少个不同的前缀，编号从$1$开始。

思路：

对于第$i$个串，用主席树记录$0-i$个串有多少不同的前缀，这个用字典树可以方便的算出，字典树的任一个节点都代表一个前缀，所以第$i$个串新增的前缀就是$length(i)-这个串的被标记过的节点$。
先把所有前缀都当做新出现的，计数会增加$length(i)$，然后再减去出现过的。
具体操作就是在字典树中沿着串走，发现已存在的节点就更新标记，计数-1。
查询时$[l,r]$就在$第r棵主席树$中求$[l,r]$区间和就行了。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+5;
string s[N];
int n;
struct Trie{
    int son[N][26], end[N];
    int root, alloc;
    int newnode(){
        memset(son[alloc], -1, sizeof(son[alloc]));
        return alloc++;
    }
    void init(){
        alloc = 0;
        root = newnode();
        memset(end, 0, sizeof(end));
    }
    void insert(string& s){
        int p = root;
        for(int i = 0; i < s.length(); ++i){
            int nx = s[i]-'a';
            if(son[p][nx] == -1) son[p][nx] = newnode();
            p = son[p][nx];
        }
    }
}trie;

int root[N<<5], ls[N<<5], rs[N<<5], val[N<<5], alloc;
void build(int &rt, int l, int r){
    rt = alloc++;
    val[rt] = 0;
    if(l == r) return;
    int mid = (l+r) >> 1;
    build(ls[rt], l, mid);
    build(rs[rt], mid+1, r);
}
inline void push_up(int rt){
    val[rt] = val[ls[rt]] + val[rs[rt]];
}
void update(int u, int &v, int l, int r, int x, int nval){
    v = alloc++;
    ls[v] = ls[u], rs[v] = rs[u];
    if(l == r){
        val[v] = val[u] + nval;
        return;
    }
    int mid = (l+r) >> 1;
    if(x <= mid) update(ls[u], ls[v], l, mid, x, nval);
    else update(rs[u], rs[v], mid+1, r, x, nval);
    push_up(v);
}
int query(int rt, int l, int r, int ql, int qr){
    if(ql <= l && qr >= r) return val[rt];
    int res = 0, mid = (l+r) >> 1;
    if(ql <= mid) res += query(ls[rt], l, mid, ql, qr);
    if(qr > mid) res += query(rs[rt], mid+1, r, ql, qr);
    return res;
}

void init(){
    alloc = 0;
    trie.init();
    for(int i = 1; i <= n; ++i){
        cin >> s[i];
        trie.insert(s[i]);
    }
    build(root[0], 1, n);
    for(int i = 1; i <= n; ++i){
        int p = trie.root;
        update(root[i-1], root[i], 1, n, i, s[i].size());
        for(int j = 0; j < s[i].length(); ++j){
            int nx = s[i][j]-'a';
            p = trie.son[p][nx];
            if(trie.end[p]) update(root[i], root[i], 1, n, trie.end[p], -1);
            trie.end[p] = i;
        }
    }
}

int main(){
    while(scanf("%d", &n) != EOF){
        init();
        int q;
        scanf("%d", &q);
        int z = 0;
        while(q--){
            int l, r;
            scanf("%d%d", &l, &r);
            int ql = min((l+z)%n+1, (r+z)%n+1), qr = max((l+z)%n+1, (r+z)%n+1);
            z = query(root[qr], 1, n, ql, qr);
            printf("%d\n", z);
        }
    }
}