Om Nom and Necklace's 题解

听说这题今天刚传上本OJ，那我就来勇一发吧

题意：给一个字符串，问该字符串的 $[1,i]$ 位上的字符串能不能由A+B+A+B+…+A构成？其中 $k+1$ 个A,$k$ 个B,A,B可以为空串。

看到题面，先想到可以分别暴力枚举A,B，但显然不太好做，换思路，emmm…

我们可以把AB拼起来看成一整个字符串，那原题就变成了检查能否用 $k$ 个AB和一个A拼成，那么首先求个最小循环节长度，用kmp的next数组得到。

最小循环节的整数倍（设为 $tmp$）还是循环节，记作 $cir$,那么问题就是判断能否存在 $tmp$ 使得 $i/(tmp\ast cir)=k$ 或 $k+1$，于是只要把 $tmp$ 用 $i,cir,k$ 表示，检验 是否满足条件即可

上code

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n,k,cnt=0,nxt[N];
char s[N];
void freo(){freopen("omnom.in","r",stdin); freopen("omnom.out","w",stdout);} 
void init(){scanf("%d%d",&n,&k); scanf("%s",s+1);}
void work(){
	for(int i=2;i<=n;++i){
		while(cnt&&s[cnt+1]!=s[i]) cnt=nxt[cnt]; if(s[cnt+1]==s[i]) ++cnt; nxt[i]=cnt;
	}
}
int che(int now){
	int cir=now-nxt[now],tmp=now/(k*cir);
	if((now%(k+1)==0&&(now/(k+1))%cir==0)||(tmp>0&&now/(cir*tmp)==k)) return 1;
	return 0;
}
void prin(){for(int i=1;i<=n;++i) if(che(i)) printf("1"); else printf("0");}
int main(){freo(); init(); work(); prin(); return 0;}

就这么。。。完了啊。。。