POJ 2406 Power Strings

一道用到KMP的题目

做题首先需要一个结论（可以自己证）

如果len%(len-nxt[len])==0 ，答案是len/(len-nxt[len])；

否则是1。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=1000000;
inline int read(){
	int x=0,f=1,ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
char s[MAXN];
int len;
int nxt[MAXN];
inline void get_nxt(){
	nxt[0]=-1;
	int i=0,k=-1;
	while(i<len){
		if(k==-1||s[i]==s[k]) nxt[++i]=++k;
		else k=nxt[k];
	}
}
int main(){
	scanf("%s",s);len=strlen(s);
	while(len!=1||s[0]!='.'){
		get_nxt();
		if(len%(len-nxt[len])==0) printf("%d\n",len/(len-nxt[len]));
		else puts("1");
		scanf("%s",s);len=strlen(s);
	}
	return 0;
}