CodeForces 235C（后缀自动机）

(http://acm.hust.edu.cn/vjudge/contest/view.action?cid=105904#problem/F)
题意：给定串s，Q组查询字符串a，问s有多少子串可由a循环构成。
解法：对s构造后缀自动机，将每个a复制成aa进行查询，要注意，s的每个子串可能和a的很多循环串相同，但只能算一次。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <queue>
#include <tr1/unordered_set>
//#include <tr1/unordered_map>
#include <bitset>
//#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

#define lson l, m, ls[rt]
#define rson m+1, r, rs[rt]
#define inf 1e9
#define debug(a) cout << #a" = " << (a) << endl;
#define debugarry(a, n) for (int i = 0; i < (n); i++) { cout << #a"[" << i << "] = " << (a)[i] << endl; }
#define clr(x, y) memset(x, y, sizeof x)
#define ll long long
#define ull unsigned long long
#define FOR(i,a,b) \
    for(i=a;a<b?i<=b:i>=b;a<b?i++:i--)

const int maxn=2e6+30;

struct suffix_auotmation{
    int son[maxn][26],pre[maxn],step[maxn];
    int last,si;
    void pushback(int v)
    {
        step[++si]=v;
        pre[si]=-1;
        clr(son[si],-1);
    }
    void Extend(int ch)
    {
        pushback(step[last]+1);
        int p=last,np=si;
        while(p!=-1&&son[p][ch]==-1)
            son[p][ch]=np,p=pre[p];
        if(p==-1) pre[np]=0;
        else{
            int q=son[p][ch];
            if(step[q]==step[p]+1) pre[np]=q;
            else{
                pushback(step[p]+1);
                int nq=si;
                memcpy(son[nq],son[q],sizeof son[q]);
                pre[nq]=pre[q];
                pre[q]=pre[np]=nq;
                while(p!=-1&&son[p][ch]==q)
                    son[p][ch]=nq,p=pre[p];
            }
        }
        last=np;
    }
    void init()
    {
        si=last=0;
        clr(son[0],-1);
        pre[0]=-1;
        step[0]=0;
    }

    int right[maxn],pos[maxn];
    int cnt[maxn];
    void getright(char *s)
    {
        clr(cnt,0);
        for(int i=0;i<=si;i++)
            cnt[step[i]]++;
        for(int i=1;i<=si;i++)
            cnt[i]+=cnt[i-1];
        for(int i=0;i<=si;i++)
            pos[--cnt[step[i]]]=i;
        int p=0;
        for(int i=0;i<=si;i++) right[i]=0;
        for(int i=0;s[i];i++)
        {
            p=son[p][s[i]-'a'];
            right[p]++;
        }
        for(int i=si;i>=0;i--) if(~pre[pos[i]])
            right[pre[pos[i]]] += right[pos[i]];
        right[0]=0;
    }
    ll solve(char *s,int len)
    {
        ll ans=0;
        int p=0;
        set<int>Q;
        int l=0;
        for(int i=0;s[i];i++)
        {
            int c=s[i]-'a';
            while( p!=-1 && son[p][c]==-1 )
                p=pre[p],l=step[p];
            if(p==-1) p=0;
            else{
                p=son[p][c];
                l++;
                while( pre[p]!=-1 && step[pre[p]] >= len )
                    p=pre[p],l=step[p];
                if( l >= len ){
                    Q.insert(p);
                    //printf("i %d l %d p %d right %d\n",i,l,p,right[p]);
                }
            }
        }
        set<int>::iterator it;
        for(it=Q.begin();it!=Q.end();it++)
            ans += (ll)right[*it];
        return ans;
    }
}su;

char s[maxn],ss[maxn];

int main()
{
//  freopen("input.txt","r",stdin);
    //freopen("out2.txt","w",stdout);
    int n;
    while(~scanf("%s",s))
    {
        su.init();
        for(int i=0;s[i];i++)
            su.Extend(s[i]-'a');
        su.getright(s);
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%s",s);
            int len=strlen(s);
            strcpy(ss,s);
            strcat(s,ss);
//          debug(s);
            ll ans=su.solve(s,len);
            printf("%I64d\n",ans);
        }
    }
    return 0;
}