ZeptoLab Code Rush 2015 D. Om Nom and Necklace（KMP+数学）

题目链接


思路：如果要分别讨论A和B的情况的话有点复杂，我们可以合并一下AB，如果合并AB能形成循环周期为k的话就是1，否则要判断一下t/k-t%k是否大于0，其中t是循环周期。这里说一下kmp的最小循环节就是len-next【len】。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+1;
int n,k,Next[maxn],ans[maxn];
char s[maxn]; 
void getNext()
{
	int j=-1;
	Next[0]=-1;
	for(int i=1;i<n;++i)
	{
		while(j!=-1&&s[i]!=s[j+1]) j=Next[j];
		if(s[i]==s[j+1]) j++;
		Next[i]=j;
	}
}
int main()
{
	scanf("%d%d",&n,&k);
	scanf("%s",s);
	getNext();
	for(int i=0;i<n;++i)
	{
		int d=i-Next[i];
		if((i+1)%d==0) {
			int t=(i+1)/d;//循环周期 
			if(t/k-t%k>=0) ans[i]=1;
		}
		else
	{
		int t=(i+1)/d;
			if(t/k-t%k>0) ans[i]=1;//B串必须存在 
	 } 
	}
	for(int i=0;i<n;++i) printf("%d",ans[i]);
 }