[BZOJ3172][Tjoi2013]单词

原题地址

我们把所有单词”拼”起来成为一个串，两个单词之间用分隔符隔开(我的代码里是’{‘)，那么要求的就是每个单词在这个拼起来的串中的出现次数。

那么这个就是后缀自动机统计子串出现次数的经典问题。

一个状态的Right集大小就是以它为起点，以一个终态为终点的路径条数(如果这个点本身就是终态，那么这一个点也算一条合法路径)，那么统计这个可以用拓扑排序后DP解决(从FHQ博客上看的)，我用的是记忆化搜索，感觉好写一点…

貌似还可以用Parent树来搞，不过我太弱了不会写OLZ…

写的时候忘记更新nq节点的nq->l了，导致出错，下次一定要注意.

AC code:

#include <cstdio>#include <cstring>const int K=27;const int N=210;const int L=2100010;int  n,cnt,ls;int  len[N],head[N];bool vis[L];char s[L],t[L];struct nod{    int l,w,num;    nod *pr,*ch[K];}pool[L];struct SAM{    nod *root,*last;    SAM(){        root=last=&pool[0];    }    void extend(int x){        nod *p=last,*np=&pool[++cnt];        for(last=np,np->num=cnt,np->l=p->l+1;p&&(!p->ch[x]);p->ch[x]=np,p=p->pr) ;        if(!p) np->pr=root;        else{            nod *q=p->ch[x];            if(p->l+1==q->l) np->pr=q;            else{                nod *nq=&pool[++cnt];                *nq=*q;nq->l=p->l+1;nq->num=cnt;q->pr=np->pr=nq;                for(;p&&p->ch[x]==q;p->ch[x]=nq,p=p->pr) ;            }        }    }    void dfs(nod *p){        vis[p->num]=1;        for(int i=0;i<K;i++){            if(!p->ch[i]) continue;            if(!vis[p->ch[i]->num]) dfs(p->ch[i]);            p->w+=p->ch[i]->w;        }    }    void build(){        for(nod *i=last;i!=root;i=i->pr) i->w=1;        dfs(root);    }    void match(int L,int R){        nod *p=root;        for(int i=L;i<=R;p=p->ch[s[i++]-'a']) ;        printf("%d\n",p->w);    }}AM;int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++){        scanf("%s",t);        len[i]=strlen(t);        strcat(s,t);        int l=strlen(s);        s[l]='{';        s[l+1]='\0';        head[i+1]=l+1;    }    ls=strlen(s);    for(int i=0;i<ls;i++) AM.extend(s[i]-'a');    AM.build();    for(int i=1;i<=n;i++) AM.match(head[i],head[i]+len[i]-1);    return 0;}