hihocoder#1465 : 后缀自动机五·重复旋律8

Time Limit:10000ms

Case Time Limit:1000ms

Memory Limit:256MB

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一段音乐旋律可以被表示为一段数构成的数列。

小Hi发现旋律可以循环，每次把一段旋律里面最前面一个音换到最后面就成为了原旋律的“循环相似旋律”，还可以对“循环相似旋律”进行相同的变换能继续得到原串的“循环相似旋律”。

小Hi对此产生了浓厚的兴趣，他有若干段旋律，和一部音乐作品。对于每一段旋律，他想知道有多少在音乐作品中的子串（重复便多次计）和该旋律是“循环相似旋律”。

解题方法提示

输入

第一行，一个由小写字母构成的字符串S，表示一部音乐作品。字符串S长度不超过100000。

第二行，一个整数N，表示有N段旋律。接下来N行，每行包含一个由小写字母构成的字符串str，表示一段旋律。所有旋律的长度和不超过 100000。

输出

输出共N行，每行一个整数，表示答案。

Sample Input

abac
3
a
ab
ca

Sample Output

2
2
1

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>

const int MAXN = 2*1e5+5;//空间多开点。 

struct Suffix_Automaton
{
    int las,tot,fa[MAXN],son[MAXN][26],len[MAXN],A[MAXN],tm[MAXN],right[MAXN],vis[MAXN];
    char s[MAXN];

    Suffix_Automaton() {tot=las=1;}
    void Init(){
		memset(fa,0,sizeof fa);
		memset(son,0,sizeof son);
		memset(len,0,sizeof len);
		memset(A,0,sizeof A);
		memset(tm,0,sizeof tm);
		memset(right,0,sizeof right);
		memset(vis,0,sizeof vis);
		tot = las = 1;
	}
    void Insert(int c)
    {
        int np=++tot,p=las;
        len[las=np]=len[p]+1, right[np]=1;
        for(; p&&!son[p][c]; p=fa[p]) son[p][c]=np;
        if(!p) fa[np]=1;
        else
        {
            int q=son[p][c];
            if(len[q]==len[p]+1) fa[np]=q;
            else
            {
                int nq=++tot; len[nq]=len[p]+1;
                memcpy(son[nq],son[q],sizeof son[q]);
                fa[nq]=fa[q], fa[q]=fa[np]=nq;
                for(; son[p][c]==q; p=fa[p]) son[p][c]=nq;
            }
        }
    }
    void Build()
    {
        scanf("%s",s+1); int l=strlen(s+1);
        for(int i=1; i<=l; ++i) Insert(s[i]-'a');
        for(int i=1; i<=tot; ++i) ++tm[len[i]];
        for(int i=1; i<=l; ++i) tm[i]+=tm[i-1];
        for(int i=1; i<=tot; ++i) A[tm[len[i]]--]=i;
        for(int i=tot,x=A[i]; i; x=A[--i]) right[fa[x]]+=right[x];
    }
    int Calc(int p,int l,int id)
    {
        for(; len[fa[p]]>=l; p=fa[p]);
        return vis[p]==id?0:(vis[p]=id,right[p]);
    }
    void Query()
    {
        static int Index=0;
        ++Index;
        scanf("%s",s+1); int l=strlen(s+1);
        for(int i=1; i<=l; ++i) s[i+l]=s[i];
        int ans=0;
        for(int c,now=0,p=1,i=1; i<=l<<1; ++i)
        {
            if(son[p][c=s[i]-'a']) p=son[p][c], ++now;
            else
            {
                for(; p&&!son[p][c]; p=fa[p]);
                if(!p) p=1, now=0;
                else now=len[p]+1, p=son[p][c];
            }
            if(now>=l) ans+=Calc(p,l,Index);
        }
        printf("%d\n",ans);
    }
}sam;

int main()
{
    sam.Build();
    int n; 
	scanf("%d",&n);
    while(n--) sam.Query();
    
    return 0;
}