POJ 2406 Power Strings（KMP求最小循环节）

传送门

用next数组求出整个数组的最大前缀，如果整个串是用循环节组成的，那么 n - next[n] 也就是最小循环节，验证最小循环节会被n整出。

#include<iostream>
#include<cstdio>
#include<cstring>
const int maxn = 1e6+10;
int nt[maxn];
using namespace std;
void Kmp_pre(char s[])
{
    int i = 0,j = -1;
    nt[0] = -1;
    int len = strlen(s);
    while(i < len)
    {
        if(-1 == j || s[i] == s[j])
        {
            i++;
            j++;
            nt[i] = j;
        }
        else
            j = nt[j];
    }
}
int main()
{
    char s[maxn];
    while(scanf("%s",s) && s[0] != '.')
    {
        Kmp_pre(s);
        int len = strlen(s);
        int m = nt[len];
        if(len % (len - m))
            printf("1\n");
        else
            printf("%d\n",len / (len - m));
    }
    return 0;
}