【题解】Rusty String [CF827E]
【题解】Rusty String [CF827E]
传送门：\(\text{Rusty String}\) \(\text{[CF827E]}\)
【题目描述】
多组数据，每组数据给出一个由 \(V,K,?\) 三种字符组成的字符串，其中 \(?\) 可以替换为 \(V,K\) 中的任意一个，求替换后的字符串可以存在的循环周期有哪些。
【分析】
乍一看像是带通配符的字符串匹配（不会的强烈建议去康康这个），由于字符集不大，按照套路就应该枚举两个字符 \(a \ne b\in\{V,K\}\)，然后设 \(f(i)=[A[i]=a],\) \(g(i)=[A[i]=b]\)
则有： \(PA(k)=\sum_{i=1}^{n}f(i)g(i+k)\)，翻转 \(f\) 得到：\(PA(k)=\sum_{i=1}^{n}f(n-i+1)g(i+k)\) \(=\sum_{i+j=n+k+1}f(i)g(j)\)
很明显是个 \(\text{FFT}\) 的板子，把式子卷起来就可以了。
按照以往的“经验”，或者说惯性思维，多半会这样做（比如说我），但写出来却发现酱紫连第一个样例都过不了。
那么问题出在哪儿了呢？
注意：当 \(PA(k) \ne 0\) 时说明在某个地方出现了 \('V'\) 与 \('K'\) 相对应的情况，以 \(k\) 作为周期肯定不合法，但 \(PA(k) = 0\) 并不能说明 \(k\) 一定是合法周期，因为这道题中的特殊符号不是通配符，而是未知字符，通配符可以任意匹配，但未知字符只能匹配一种。
比如样例 \(1\)，用上面的错误算法多输出了一个 \(2\)，在这种情况下 \(S_3('?')\) 分别与 \(S_1('V'),S_5('K')\) 进行了匹配，因此以 \(2\) 作为周期是不合法的。
考虑改进算法：
之前的做法错在算出来的式子只进行了一周目的匹配判断，那么我们只要将二周目、三周目……全部都判断一下就好了呀，正巧序列 \(PA\) 完美的包含了所有长度的循环情况，所以直接暴力枚举周期的倍数即可（复杂度为 \(O(\frac{n}{1}+\frac{n}{2}+\frac{n}{3} \cdots \frac{n}{n})=O(nlogn)\)）。
总时间复杂度为：\(O(nlogn)\) 。
你说啥？卡常？玄学优化减少调用次数？不存在的，我写了这么久的 \(\text{NTT}\)，还从未被卡过（小骄傲）。
\(\text{emm...}\)
等等，还没完，仔细看题啊：“保证输入字符串总长不超过 \(5∗10^5\) 。” 注意是输入总长，所以不要偷懒用 \(\text{memset}\)！
【Code】
#include<algorithm>
#include<cstring>
#include<cstdio>
#define LL long long
#define Re register int
using namespace std;
const int N=1048576+3,P=998244353,G=3;
int n,T,ans,invG,f[N],g[N],tr[N],PA[N],can[N];char A[N];
inline void in(Re &x){
int f=0;x=0;char c=getchar();
while(c<'0'||c>'9')f|=c=='-',c=getchar();
while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
x=f?-x:x;
}
inline int mi(Re x,Re k){
Re s=1;
while(k){
if(k&1)s=(LL)s*x%P;
x=(LL)x*x%P,k>>=1;
}
return s;
}
inline int inv(Re x){return mi(x,P-2);}
inline void NTT(Re *f,Re n,Re op){
for(Re i=0;i<n;++i)if(i<tr[i])swap(f[i],f[tr[i]]);
for(Re p=2;p<=n;p<<=1){
Re len=p>>1,w1=mi(op?invG:G,(P-1)/p);
for(Re st=0;st<n;st+=p)
for(Re j=st,base=1;j<=st+len-1;++j){
Re tmp=(LL)base*f[j+len]%P;
f[j+len]=(f[j]-tmp+P)%P,(f[j]+=tmp)%=P;
base=(LL)base*w1%P;
}
}
}
inline void sakura(Re *f,Re n,Re *g,Re m){//卷卷卷
Re n_=n;
for(m+=n,n=1;n<=m;n<<=1);Re invn=inv(n);
for(Re i=n_+1;i<=n;++i)f[i]=g[i]=0;//注意初始化
for(Re i=1;i<n;++i)tr[i]=(tr[i>>1]>>1)|((i&1)?n>>1:0);
NTT(f,n,0),NTT(g,n,0);
for(Re i=0;i<n;++i)f[i]=(LL)f[i]*g[i]%P;
NTT(f,n,1);
for(Re i=0;i<=m;++i)f[i]=(LL)f[i]*invn%P;
}
int main(){
// freopen("123.txt","r",stdin);
in(T),invG=inv(G);
while(T--){
in(n),scanf("%s",A+1),ans=0;
for(Re i=1;i<=n;++i)PA[i]=0;//注意初始化
for(Re i=1;i<=n;++i)f[n-i+1]=(A[i]=='K'),g[i]=(A[i]=='V');
sakura(f,n,g,n);
for(Re i=1;i<n;++i)PA[i]+=f[i+n+1];//PA[n]可以不用管，因为一定合法
for(Re i=1;i<=n;++i)f[n-i+1]=(A[i]=='V'),g[i]=(A[i]=='K');
sakura(f,n,g,n);
for(Re i=1;i<n;++i)PA[i]+=f[i+n+1];
for(Re i=1;i<=n;++i){
can[i]=1;
for(Re j=i;j<=n&&can[i];j+=i)can[i]&=(!PA[j]);//枚举倍数依次判断
ans+=can[i];
}
printf("%d\n",ans);
for(Re i=1;i<=n;++i)if(can[i])printf("%d ",i);puts("");
}
}
