【CodeForces

题意

  给出一个字符串s1和q个询问，每个询问给出一个字符串s2，问这个询问的字符串的所有不同的周期串在s1中出现的次数的和。

分析

  对于s1建后缀自动机。对于询问的每个字符串s2，我们按照处理循环串的方法，将它长度乘二再复制一遍。然后根据s2在自动机上跑，当长度len=n的时候，就更新答案。因为要求统计的是不同的周期串，所以对于每个状态都需要打一个vis标记。

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

using namespace std;
const int maxn=2e6+10;
char s[maxn];
struct state{
    int len,link;
    int next[26];
}st[2*maxn];
int last,cur,sz,Q,n;
int cnt[2*maxn],c[2*maxn],vis[2*maxn];
void init(){
    sz=1;
    last=cur=0;
    st[0].link=-1;
    st[0].len=0;
}
void build_sam(int c){
    cur=sz++;
    st[cur].len=st[last].len+1;
    cnt[cur]=1;
    int p;
    for(p=last;p!=-1&&st[p].next[c]==0;p=st[p].link){
        st[p].next[c]=cur;
    }
    if(p==-1)
        st[cur].link=0;
    else{
        int q=st[p].next[c];
        if(st[q].len==st[p].len+1)
            st[cur].link=q;
        else{
            int clone=sz++;
            st[clone].len=st[p].len+1;
            st[clone].link=st[q].link;
            for(int i=0;i<26;i++)
                st[clone].next[i]=st[q].next[i];
            for(;p!=-1&&st[p].next[c]==q;p=st[p].link){
                st[p].next[c]=clone;
            }
            st[cur].link=st[q].link=clone;
        }
    }
    last=cur;
}
int cmp(int a,int b){
    return st[a].len>st[b].len;
}
int solve(int id){
    int res=0;
    int u=0,len=0;
    for(int i=0;i<2*n-1;i++){
        int c=s[i]-'a';
        while(u!=-1&&(st[u].next[c]==0))
            u=st[u].link,len=st[u].len;
        if(u==-1)
            u=0,len=0;
        else{
            u=st[u].next[c];
            len++;
            if(len>=n&&vis[u]!=id){
                res+=cnt[u];
                vis[u]=id;
            }
            while(n!=1&&st[u].link!=-1&&st[st[u].link].len>=n-1)
                u=st[u].link,len=st[u].len;
        }
    }
    return res;
}

int main(){
    scanf("%s",s);
    n=strlen(s);
    init();
    for(int i=0;i<n;i++)
        build_sam(s[i]-'a');
    for(int i=0;i<sz;i++)
        c[i]=i;
    sort(c,c+sz,cmp);
    for(int i=0;i<sz;i++){
        int o=c[i];
        if(st[o].link!=-1){
            cnt[st[o].link]+=cnt[o];
        }
    }

    scanf("%d",&Q);
    for(int i=1;i<=Q;i++){
       // memset(vis,0,sizeof(vis));
        scanf("%s",s);
        n=strlen(s);
        for(int j=0;j<n;j++)
            s[j+n]=s[j];
        int res=solve(i);
        printf("%d
",res);
    }
return 0;
}