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
*/