POJ2406 Power Strings【KMP】

题目链接：

http://poj.org/problem?id=2406

题目大意：

给定两个字符串a和b，定义a*b为两个字符串的链接。比如，a = "abc"，b = "def"，则

a*b ="abcdef"。这个定义当作是多项式。则一个字符串的非负整数次幂可定义如下：

a^0 = ""，a^(n+1) = a*a^n。

现在给你一个字符串s，求出最大的n，满足s = a^n(a为s的某个子串)。比如s = "aaaa"，

则n最大为4，a = "a"，s = "a"^4。

思路：

对于给定的字符串s，最短的重复子串a是s[Next[len]] s[Next[len+1]] … s[len-1]。当

len % Next[len] = 0时，说明字符串s有不是它本身重复子串a，最大的n为len/Next[len]。

当len % Next[len] != 0时，除了s，没有重复子串a满足要求，这时n = 1。

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

char str[1000010];
int Next[1000010],len;

void GetNext()
{
    int i = 0,j = -1;
    Next[0] = -1;
    while(i <= len)
    {
        if(j == -1 ||  str[i]==str[j])
        {
            i++,j++;
            Next[i] = j;
        }
        else
            j = Next[j];
    }
}
int main()
{
    while(cin >> str && strcmp(str,".")!=0)
    {
        len = strlen(str);
        GetNext();
        if(len % (len-Next[len]) == 0)
            cout << len/(len-Next[len]) << endl;
        else
            cout << 1 << endl;
    }

    return 0;
}