字符串专项：CodeForces 235C

一道后缀自动机的算是模板题吧，可是我还是折腾了一个下午……总算对这个模板有了稍微深入的理解，debug了很久，也发现了不少的问题。现在算是基本会用后缀数组了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int maxn=2000010;
const int sigma=26;
int n,sz;
struct state
{
    state *pre,*go[sigma];
    int val,cnt,flag;
    void clear()
    {
        pre=0;val=0;cnt=0;flag=0;
        memset(go,0,sizeof(go));
    }
} *root,*last,st[maxn],*r[maxn];
int t[maxn];
void init()
{
    sz=0;
    root=last=&st[sz++];
    root->clear();
}
void extend(int w)
{
    state *p=last;
    state *np=&st[sz++];
    np->clear();
    np->val=p->val+1;
    np->cnt=1;
    while(p && p->go[w]==0)
    {
        p->go[w]=np;
        p=p->pre;
    }
    if(p==0) np->pre=root;
    else
    {
        state *q=p->go[w];
        if(q->val==p->val+1) np->pre=q;
        else
        {
            state *nq=&st[sz++];
            *nq=*q;
            nq->val=p->val+1;
            nq->cnt=0;
            q->pre=nq;
            np->pre=nq;
            while(p && p->go[w]==q)
            {
                p->go[w]=nq;
                p=p->pre;
            }
        }
    }
    last=np;
}
char s[maxn];
vector<int> v[maxn];
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%s",s);
    int m=strlen(s);
    init();
    for(int i=0;i<m;i++) extend(s[i]-'a');
//    for(int i=1;i<sz;i++) v[st[i].val].push_back(i);
//    for(int i=m;i>=1;i--)
//        for(int j=0;j<v[i].size();j++)
//        {
//            int x=v[i][j];
//            if(st[x].pre) st[x].pre->cnt+=st[x].cnt;
//        }
    memset(t,0,sizeof(t));
    for(int i=1;i<sz;i++) t[st[i].val]++;
    for(int i=1;i<=m;i++) t[i]+=t[i-1];
    for(int i=1;i<sz;i++) r[t[st[i].val]--]=&st[i];
    for(int i=sz-1;i>0;i--)
    {
        //cout<<r[i]->cnt<<" "<<r[i]->val<<endl;
        r[i]->pre->cnt+=r[i]->cnt;
    }
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s);
        m=strlen(s);
        for(int j=0;j<m;j++) s[j+m]=s[j];
        s[2*m]=0;
        state *p=root;
        int len=0,ans=0;
        for(int j=0;j<2*m;j++)
        {
            int w=s[j]-'a';
            while(p && p->go[w]==0)
            {
                p=p->pre;len=p?p->val:0;
            }
            if(p==0)
            {
                p=root;
                len=0;
                continue;
            }
            p=p->go[w];len++;
            if(len>=m)
            {
                state *tmp=p;
                while(tmp->pre && tmp->pre->val>=m)
                    tmp=tmp->pre;
                if(tmp->flag!=i)
                {
                    tmp->flag=i;
                    //cout<<j<<" "<<tmp->val<<" "<<tmp->cnt<<endl;
                    ans+=tmp->cnt;
                }
            }
            //cout<<j<<" "<<ans<<endl;
        }
        printf("%d\n",ans);
    }
    return 0;
}