Hdu 5790 Prefix(每次查询问你第L个字符串和第R个字符串之间的公共前缀有多少个)

传送门:Hdu 5790 Prefix

题意:给你n个字符串（1≤N≤100000),字符串的总长度小于等于100000
接下来有q个查询(1<=q<=100000),每次查询问你第L个字符串和第R个字符串之间的公共前缀有多少个

思路:将每个前缀都表示成在字典树上的值,然后插入到主席树中->主席树区间内有多少个不同的值

#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
int a[maxn],t[maxn],net[maxn],root[maxn],sz;
struct node{
    int l,r,w;
}T[maxn*40];

void update(int &i,int l,int r,int num,int v){
    T[++sz]=T[i],i=sz;
    T[i].w+=v;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(num<=mid)
        update(T[i].l,l,mid,num,v);
    else
        update(T[i].r,mid+1,r,num,v);
}

int query(int i,int pos,int l,int r){
    if(l==pos)
        return T[i].w;
    int mid=(l+r)>>1;
    if(pos<=mid)
        return T[T[i].r].w+query(T[i].l,pos,l,mid);
    else
        return query(T[i].r,pos,mid+1,r);
}

char s[maxn];

struct Trie{
    int ch[maxn][26],sz;
    void clear(){sz=1,memset(ch[0],0,sizeof(ch[0]));}
    int idx(char x){return x-'a';}
    void insert(const char*s){
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                memset(ch[sz],0,sizeof(ch[sz]));
                ch[u][c]=sz++;
            }
            u=ch[u][c];
            a[i]=u;
        }
    }
};

Trie trie;
int st[maxn],ed[maxn];


int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        trie.clear();
        memset(net,-1,sizeof(net));
        int tot=0;
        root[0]=0,sz=0;
        for(int i=1;i<=n;i++){
            scanf("%s",s);
            trie.insert(s);
            int len=strlen(s);
            st[i]=tot+1,ed[i]=tot+len;
            for(int j=0;j<len;j++){
                tot++;
                if(net[a[j]]==-1){
                    root[tot]=root[tot-1];
                    update(root[tot],1,100000,tot,1);
                }
                else{
                    root[tot]=root[tot-1];
                    update(root[tot],1,100000,net[a[j]],-1);
                    update(root[tot],1,100000,tot,1);
                }
                net[a[j]]=tot;
            }
        }
        int L,R,Z=0,q;
        scanf("%d",&q);
        for(int i=1;i<=q;i++){
            scanf("%d%d",&L,&R);
            int tmp1,tmp2;
            tmp1=min((Z+L)%n,(Z+R)%n)+1,tmp2=max((Z+L)%n,(Z+R)%n)+1;
            Z=query(root[ed[tmp2]],st[tmp1],1,100000);
            printf("%d\n",Z);
        }
    }
    return 0;
}