Prefix
题意:
给出总长 105的N个串 ,给出 q个询问 ,查询编号 [l,r] 范围内的串有多少个不同的前缀,编号从 1 开始。
思路:
对于第
先把所有前缀都当做新出现的,计数会增加
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);
}
}
}