poj 2406 Power Strings (KMP)

# include <stdio.h>
# include <string.h>
# include <algorithm>
using namespace std;
int len;
char a[1000010];
int next[1000010];
void Getnext()
{
int i=0,j=-1;
next[0]=-1;
while(i<=len)
{
if(j==-1||a[j]==a[i])
i++,j++,next[i]=j;
else
j=next[j];
}
}
int main()
{
while(~scanf("%s",a),a[0]!='.')
{
len=strlen(a);
Getnext();
int ans=len-next[len];//最小循环节
if(len%ans)//不是最小循环节的倍数，没有循环
printf("1\n");
else
{
int flag=0;
for(int i=ans*2;i<=len;i+=ans*2)//ans*2的倍数的next必是最小循环结的倍数，可以Y几个案例
{
if(next[i]%ans)
{
flag=1;
break;
}
}
if(flag)
printf("1\n");
else
printf("%d\n",len/ans);
}

}
return 0;
}