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);
    }
}