Power Strings（poj 2406）

题意：求字符串中循环节出现的次数

KMP！！！

#include<cstdio>
#include<iostream>
#include<cstring>
#define M 1000010
using namespace std;
int fail[M],a[M],m;
char ch[M];
void kmp_init()
{
    fail[]=;
    for(int i=;i<=m;i++)
    {
        int p=fail[i-];
        while(p&&a[p+]!=a[i])p=fail[p];
        if(a[p+]==a[i])
          fail[i]=p+;
        else fail[i]=;
    }
}
int main()
{
    while(scanf("%s",ch))
    {
        if(ch[]=='.')break;
        memset(fail,,sizeof(fail));
        m=strlen(ch);
        for(int i=;i<=m;i++)
          a[i]=ch[i-];
        kmp_init();
        int ans=m-fail[m];
        if(m%ans==)printf("%d\n",m/ans);
        else printf("1\n");
    }
    return ;
}