bzoj 2434: [Noi2011]阿狸的打字机

题意：

给出一个字典树，并告诉你若干串的结束位置。求第x个串在第y个串的出现次数。

题解：

不难的AC自动机+不难的主席树。
首先对于字典树建AC自动机，那么对于一个串S，假如能在这个串的结尾点不断跳fail跳到S1的结尾点的话，就说明S的其中一个后缀是S1。那么我们可以枚举串y的前缀，假如这个前缀的后缀是x的话，ans++。即该前缀的结尾点是x结尾点在fail树上的子节点。
我们对fail树dfs序，问题就成了y串在字典树上的路径有多少个点是x结尾点的子节点，也就是在x的子树的dfs序范围内，自然想到主席树。好像有树状数组做法的
code：

#include<cstdio>#include<cstdlib>#include<cstring>#include<queue>#include<algorithm>#include<iostream>using namespace std;struct trnode{    int a[27],fail,fa,tail;}tr[100010];int tail[100010],ROOT=1,now=1,tot=1,ys[100010],z=0,n,root[100010];struct TRnode{    int lc,rc,c;}TR[2500010];struct node{    int y,next;}a[100010];int last[100010],len=0;void ins(int x,int y){    a[++len].y=y;    a[len].next=last[x];last[x]=len;}char ch;int insert(int x,int c){    if(!tr[x].a[c])    {        tr[x].a[c]=++tot;tr[tot].fa=x;    }    return tr[x].a[c];}queue<int> q;void make_fail(){    q.push(ROOT);    while(!q.empty())    {        int x=q.front();q.pop();        for(int i=0;i<26;i++)        {            if(!tr[x].a[i]) continue;            if(x==ROOT) tr[tr[x].a[i]].fail=ROOT;            else            {                int j=tr[x].fail;                while(j!=ROOT&&!tr[j].a[i]) j=tr[j].fail;                tr[tr[x].a[i]].fail=tr[j].a[i];            }            q.push(tr[x].a[i]);        }    }}void dfs(int x){    ys[x]=tr[x].tail=++z;    for(int i=last[x];i;i=a[i].next)    {        int y=a[i].y;        dfs(y);        tr[x].tail=tr[y].tail;    }}void update(int &x,int froot,int l,int r,int k,int c){    x=++tot;    TR[x].lc=TR[froot].lc;TR[x].rc=TR[froot].rc;TR[x].c=TR[froot].c;    if(l==r){TR[x].c+=c;return;}    int mid=(l+r)/2;    if(k<=mid) update(TR[x].lc,TR[froot].lc,l,mid,k,c);    else update(TR[x].rc,TR[froot].rc,mid+1,r,k,c);    TR[x].c=TR[TR[x].lc].c+TR[TR[x].rc].c;}int findans(int x,int l,int r,int fl,int fr){    if(!x) return 0;    if(l==fl&&r==fr) return TR[x].c;    int mid=(l+r)/2;    if(fr<=mid) return findans(TR[x].lc,l,mid,fl,fr);    if(fl>mid) return findans(TR[x].rc,mid+1,r,fl,fr);    return findans(TR[x].lc,l,mid,fl,mid)+findans(TR[x].rc,mid+1,r,mid+1,fr);}void bt(int x){    update(root[x],root[tr[x].fa],1,n,ys[x],1);    for(int i=0;i<26;i++)        if(tr[x].a[i]!=0) bt(tr[x].a[i]);}int main(){    tail[1]=ROOT;    for(int i=0;i<26;i++) tr[ROOT].a[i]=++tot,tr[tot].fa=ROOT;    while(scanf("%c",&ch))    {        if(ch=='\n') break;        else if(ch=='P') now++,tail[now]=tail[now-1];        else if(ch=='B') tail[now]=tr[tail[now]].fa;        else tail[now]=insert(tail[now],ch-'a');    }    make_fail();    for(int i=2;i<=tot;i++) ins(tr[i].fail,i);    n=tot;tot=0;    dfs(1);bt(ROOT);    int T;scanf("%d",&T);    while(T--)    {        int x,y;scanf("%d %d",&x,&y);        if(x>=now||y>=now){printf("0\n");continue;}        x=tail[x];y=tail[y];int ans=0;        printf("%d\n",findans(root[y],1,n,ys[x],tr[x].tail));    }}

数据生成器：

#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<ctime>using namespace std;int main(){    srand(time(0));    int n=rand()%1000+1,num=0,t=1,k=0;    while(t<=n)    {        int tmp=rand()%5;        if(tmp==0)        {            if(num==0) continue;            else printf("B"),num--,t++;        }        else             if(tmp==1)            {                if(num==0) continue;                printf("P"),t++,k++;            }        else printf("%c",rand()%2+'a'),num++,t++;    }    printf("\n");    int m=rand()%30+1;    printf("%d\n",m);if(!k) k=1;    for(int i=1;i<=m;i++)    {        int x=rand()%k+1,y=rand()%k+1;        printf("%d %d\n",x,y);    }}