[CF235C] Cyclical Quest

题目大意

给一个长度为n 的字符串s，每次询问一个串q 问s 有多少个子串与q 循环同构。
s长度,q总长度小于10^6。

分析

就是把q复制一遍，然后再sam上面跑，如果循环了，就不加入答案。
好久没打SAM…需要复习一下。
卡了好久，因为有几句话打反了。
为了程序不出错，最好把root设置成1。
跑的时候注意，用mat记录匹配的字符串长度，注意一定要跑到完全代表s[i~i+len(q)]的点，不然答案会小。

代码

#include<cstdio>#include<algorithm>#include<cmath>#include<cstring>#include<set>#include<bitset>using namespace std;#define fo(i,j,k) for(i=j;i<=k;i++)#define fd(i,j,k) for(i=j;i>=k;i--)#define cmax(a,b) (a=(a>b)?a:b)typedef long long ll;typedef double db;const int N=2e6+5,mo=1e8;//7707253;int tr[N][26],fail[N],mx[N],rt,kmp[N],lst[N],n,len,i,j,k,ts,q1,q2,d[N],x,tmp,l,mat,vis[N];char s[N];ll ans,cnt[N];int tt,b[N],first[N],next[N];void cr(int x,int y){    tt++;    b[tt]=y;    next[tt]=first[x];    first[x]=tt;}int ins(int lst,int x){    int np=++ts,p=lst,nq,q;    mx[np]=mx[p]+1;    cnt[np]=1;    while (p&&!tr[p][x]) tr[p][x]=np,p=fail[p];    if (!p) fail[np]=1;else    {        q=tr[p][x];        if (mx[q]==mx[p]+1) fail[np]=q;else        {            nq=++ts;            memcpy(tr[nq],tr[q],sizeof(tr[q]));            mx[nq]=mx[p]+1;            while (p&&tr[p][x]==q) tr[p][x]=nq,p=fail[p];            fail[nq]=fail[q];            fail[q]=fail[np]=nq;        }    }    return np;}int main(){    freopen("data.in","r",stdin);    //freopen("data.out","w",stdout);    scanf("%s\n",s+1);    len=strlen(s+1);    //root=1;    ts=1;    fail[1]=0;    lst[0]=1;    fo(i,1,len) lst[i]=ins(lst[i-1],s[i]-'a');    fo(i,2,ts) cr(fail[i],i);    q1=0;d[q2=1]=1;    while(q1<q2) for(int p=first[d[++q1]];p;p=next[p]) d[++q2]=b[p];    fd(i,ts,1) cnt[fail[d[i]]]+=cnt[d[i]];    scanf("%d\n",&n);    fo(l,1,n)    {        scanf("%s\n",s+1);        len=strlen(s+1);        fo(i,1,len) s[i+len]=s[i];        ans=0;tmp=1;mat=0;        fo(i,1,len*2)        {            while (tmp&&!tr[tmp][s[i]-'a']) tmp=fail[tmp],mat=mx[tmp];            if (!tmp) tmp=1,mat=0;            else tmp=tr[tmp][s[i]-'a'],mat++;            while (tmp&&mx[fail[tmp]]>=len) tmp=fail[tmp],len=mx[tmp];            if (mat>=len&&vis[tmp]!=l)  vis[tmp]=l,ans+=cnt[tmp];        }        printf("%I64d\n",ans);    }}