poj2406 Power Strings

题意：给一个字符串，找到循环节的数目。
思路：由kmp算法，next[len]知道前缀的第一个到第next[len]与后缀第len-next[len]到最后一个是相同的，如果这个字符串是循环节构成，那么len-next[len]一定是最短的循环节的长度。于是判断这个字符串是否有循环节的条件就是len%(len-next[len])==0,求出循环节的长度，总长度比上循环节长度就是个数。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<climits>
#include<stack>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
int next[1000010],len;
char s[1000010];
void get_next(char *s,int *next)
{
    memset(next,0,sizeof(next));
    next[0]=-1;
    int i=0,j=-1;
    while(i<len)
    {
        if(j==-1||s[i]==s[j])
        {
            i++,j++;
            next[i]=j;
        }
        else j=next[j];
    }
}

int main()
{
    while(scanf("%s",s)&&s[0]!='.')
    {
        len=strlen(s);
        get_next(s,next);
        if(len%(len-next[len])==0) printf("%d\n",len/(len-next[len]));
        else printf("1\n");
    }
}