hdu 5790 prefix 主席树在线维护区间不同数的个数

题意：给n个串,m次询问,每次给你L,R 要统计第L~R个串中有多少个不同的前缀。n<=10w

m<=10w 串的总长度<=10w

题解：由于是前缀，可以用trie树处理，剩下的问题就是在线访问区间内有多少个不同的数了。 我们可以记录trie上每个节点上次出现的位置。对于每个根节点i我们可以记录 前缀上次出现的位置<i的 有多少个。然后这个可以用主席树记前缀优化。

#include <bits/stdc++.h>
using namespace std;
#define maxn 110000
int a[maxn][26],triesiz,treesiz,sum[maxn*20];
int ls[maxn*20],rs[maxn*20],root[maxn],n,last[maxn];
char st[maxn];
void update(int l,int r,int x,int &y,int pos)
{
treesiz++;y=treesiz;sum[y]=sum[x]+1;
if(l==r)
{
return ;
}
ls[y]=ls[x];rs[y]=rs[x];
int mid=(l+r)>>1;
if(pos<=mid) update(l,mid,ls[x],ls[y],pos);
else update(mid+1,r,rs[x],rs[y],pos);
}
int query(int l,int r,int x,int y,int pos)
{
printf("%d %d %d %d %d\n",l,r,x,y,pos);
system("pause");
if(l==r) return sum[y]-sum[x];
int mid=(l+r)>>1;
if(pos>mid) return sum[ls[y]]-sum[ls[x]]+query(mid+1,r,rs[x],rs[y],pos);
else return query(l,mid,ls[x],ls[y],pos);
}
void ins(int num)
{
int len=strlen(st+1),t,r1=root[num-1],r2=0;
int now=0;
for(int i=1;i<=len;i++)
{
t=st[i]-'a';
if(!a[now][t])
{
triesiz++;
a[now][t]=triesiz;
now=triesiz;
last[now]=num;
update(1,n,r1,r2,1);
r1=r2;
}
else
{
now=a[now][t];
update(1,n,r1,r2,last[now]+1);
last[now]=num;
r1=r2;
}
}
root[num]=r2;
}
void cn()
{
memset(a,0,sizeof(a));
memset(ls,0,sizeof(ls));
memset(rs,0,sizeof(rs));
memset(sum,0,sizeof(sum));
memset(root,0,sizeof(root));
triesiz=treesiz=0;
}
int main()
{
while(~scanf("%d",&n))
{
cn();
for(int i=1;i<=n;i++)
{
scanf("%s",st+1);
ins(i);
}
int m,l,r,z=0;
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&l,&r);
//l = (z+l)%n+1; r = (z+r)%n+1;
//if(l > r) swap(l,r);
z=query(1,n,root[l-1],root[r],l);

printf("%d\n",z);
}
}
return 0;
}
/*
3
abc
aba
baa
3
0 2
0 1
1 1
*/