Luogu CF235C Cyclical Quest

题意：给定一个主串 \(S\) 和 \(n\) 个询问串，求每个询问串的所有循环同构在主串中出现的次数总和。

我们把 \(S\) 建出 \(SAM\) ，对每个询问串我们令 \(T'=T[1\cdots n]+T[1\cdots n-1]\) ，我们让 \(T'\) 在 \(S\) 的 \(SAM\) 上跑匹配即可。注意每个点要打一个标记，记录是否计算过贡献。

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long
#define R register int
using namespace std;
namespace Luitaryi {
inline int g() { R x=0,f=1;
  register char s; while(!isdigit(s=getchar())) f=s=='-'?-1:f;
  do x=x*10+(s^48); while(isdigit(s=getchar())); return x*f;
} const int N=2000010;
int n,m,lst=1,tot=1,tim,T;
int fa[N],len[N],c[N][26],sz[N],d[N],mem[N],vis[N];
char s[N]; ll ans;
inline void add(int ch) {
  R p=lst,np=lst=++tot;
  sz[np]=1,len[np]=len[p]+1;
  for(;p&&!c[p][ch];p=fa[p]) c[p][ch]=np;
  if(!p) fa[np]=1;
  else {
    R q=c[p][ch];
    if(len[q]==len[p]+1) fa[np]=q;
    else {
      R nq=++tot;
      memcpy(c[nq],c[q],26<<2);
      fa[nq]=fa[q],len[nq]=len[p]+1;
      fa[np]=fa[q]=nq;
      for(;p&&c[p][ch]==q;p=fa[p]) c[p][ch]=nq;
    }
  }
}
inline void solve() {
  R p=1,l=0; ans=0;
  for(R i=1;i<=m;++i) {
    R ch=s[i]-'a';
    while(p&&!c[p][ch]) p=fa[p],l=len[p];
    if(!p) p=1,l=0;
    else p=c[p][ch],++l;
    if(l>n) while(len[fa[p]]>=n) p=fa[p],l=len[p];
    if(l>=n&&vis[p]!=tim) vis[p]=tim,ans+=sz[p];
  } printf("%lld\n",ans);
}
inline void main() {
  scanf("%s",s+1),n=strlen(s+1);
  for(R i=1;i<=n;++i) add(s[i]-'a');
  for(R i=1;i<=tot;++i) ++d[len[i]];
  for(R i=1;i<=n;++i) d[i]+=d[i-1];
  for(R i=1;i<=tot;++i) mem[d[len[i]]--]=i;
  for(R i=tot;i;--i) sz[fa[mem[i]]]+=sz[mem[i]];
  for(tim=1,T=g();tim<=T;++tim) {
    scanf("%s",s+1),n=strlen(s+1),m=2*n-1;
    for(R i=1;i<n;++i) s[n+i]=s[i];
    solve();
  }
}
} signed main() {Luitaryi::main(); return 0;}

---

2020.01.10