poj 2406 Power Strings(KMP）

Power Strings（pku 2406）

给定两个字符串 a和b，我们定义 a*b为两个字符串的联接。例如，a = "abc" 且 b = "def"，则a*b = "abcdef"。把这个定义的运算看做多项式，则一个字符串的非负整数次幂定义如下： a^0 = "" (一个空字符串) 且 a^(n+1) = a*(a^n)。

输入：

每个测试样例为一行，即一个字符串s，其长度至少为1且不超过1000000。以句号“.”为最后一组测试样例。

输出：

对于每个给出的字符串s，你应该输出最大的n使得 s = a^n等于某个字符串 a。

输入样例：

abcd

aaaa

ababab

.
    输出样例：

1

4

    3

分析：

题目意思是对于给定的字符串s，找到其最短子串a，满足重复子串a可以得到字符串s。那么这个最短子串是s[next[len]]s[next[len]+1]...s[len-1]。当len%(len-next[len])==0时，说明字符串s有满足条件的不是它本身的子串，len/(len-next[len])即为所求最大的n。当len%(len-next[len])!=0时，字符串s除了它本身没有满足条件的子串，故n=1。

代码：

#include<iostream>#include<cstdio>#include<cstring>using namespace std;int next[1000005];int len;int get_next(char *p){    next[0] = -1;    int i = 0, j = -1;    while(i <= len)    {        while(j == -1 || (i < len && p[i] == p[j]) )        {            i++;            j++;            next[i] = j;        }        j = next[j];    }    return 0;}int main(){    char a[1000005];    while(scanf("%s", a) != EOF && strcmp(a,".") != 0)    {        len = strlen(a);        get_next(a);        if(len % (len - next[len]) == 0)            printf("%d\n", len / (len - next[len]));        else            printf("1\n");    }    return 0;}