Prefix HDU - 5790 字典树 + 主席树

传送门

题意：

给你$n$个串，每次询问一个区间，返回这个区间的串的不同的前缀个数，强制在线。

思路：

碰到字符串前缀的问题，我们自然的想到用字典树来解决。
对于每个串，我们将其插入字典树的时候记一下当前串的每个前缀的$id$，我们可以将问题就转化成询问区间内出现数字的个数(即出现的$id$)，因为要强制在线，我们考虑用主席树维护历史信息。
对于维护区间内出现数字的个数，这是个经典问题了，维护前缀出现的最后一个位置即可。查询的时候直接查询第$r$棵树的区间和即可。

//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=200110,mod=1e9+7,INF=0x3f3f3f3f;
const int P=131;
const double eps=1e-6;

int n,m;
int root[N],tot;
int idx,num[N*5],tree[N][26],id,st[N*5];
int pre[N*5];
char s[N];
struct Node
{
    int l,r;
    int cnt;
}tr[N*40];

int newnode()
{
    idx++;
    for(int i=0;i<26;i++) tree[idx][i]=0;
    num[idx]=0;
    return idx;
}

void add()
{
    int p=0;
    for(int i=0;s[i];i++)
    {
        int now=s[i]-'a';
        if(!tree[p][now]) tree[p][now]=newnode();
        p=tree[p][now];
        if(!num[p]) num[p]=++id;
        st[i]=num[p];
    }
}

void insert(int p,int &q,int l,int r,int pos,int x)
{
    q=++tot; tr[q]=tr[p];
    tr[q].cnt+=x;
    if(l==r) return;
    int mid=l+r>>1;
    if(pos<=mid) insert(tr[p].l,tr[q].l,l,mid,pos,x);
    else insert(tr[p].r,tr[q].r,mid+1,r,pos,x);
}

int query(int u,int l,int r,int ql,int qr)
{
    if(!u) return 0;
    if(l>=ql&&r<=qr) return tr[u].cnt;
    int ans=0,mid=l+r>>1;
    if(ql<=mid) ans+=query(tr[u].l,l,mid,ql,qr);
    if(qr>mid) ans+=query(tr[u].r,mid+1,r,ql,qr);
    return ans;
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    memset(pre,-1,sizeof(pre));
    while(scanf("%d",&n)!=EOF)
    {
        memset(tree,0,sizeof(tree));
        idx=tot=id=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s);
            add();
            root[i]=root[i-1];
            for(int j=0;s[j];j++)
            {
                if(pre[st[j]]==-1) insert(root[i],root[i],1,n,i,1);
                else
                {
                    insert(root[i],root[i],1,n,pre[st[j]],-1);
                    insert(root[i],root[i],1,n,i,1);
                }
                pre[st[j]]=i;
            }
        }
        scanf("%d",&m);
        int ans=0;
        while(m--)
        {
            int ll,rr; scanf("%d%d",&ll,&rr);
            int l=min((ll+ans)%n,(rr+ans)%n)+1;
            int r=max((ll+ans)%n,(rr+ans)%n)+1;
            printf("%d\n",ans=query(root[r],1,n,l,r));
        }
        for(int i=1;i<=id;i++) pre[i]=-1;
    }



	return 0;
}
/*

*/