[POJ 2406] Power Strings （kmp循环节）

链接

POJ 2406

---

题意

给出一个字符串，输出其最短循环节长度。

---

题解

首先说一下循环节，其实每个字符串都是有循环节的，最长的循环节就是它本身。
如果一个字符串s可以写成某个字符串d复制k次得到的，那么d就是s的循环节，最短循环节就是d中最小的一个。

其实这题是很容易解的，我们对该字符串s求next数组，相当于得到了所有与s后缀匹配的真前缀，一一检测即可。
但是总结这题主要是为了一个结论，其实无需求出所有真前缀，只要验证最长真前缀next[n]是否与len[s]周期相同即可：
if(len % (len - next[n]) == 0) …;
可以看出如果next[n]大于len的一半，那么s的循环节就是它本身了。
这样得到了一个结论：
如果与字符串s后缀匹配的最长真前缀与s同周期，则该周期为s的最短循环节，否则s的循环节为len[s]。

---

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn (1000010)
int next[maxn];
void pre_kmp(char s[], int n)
{
    memset(next, 0, sizeof(int)*(n+1));
    for(int i = 1, j; i < n; i++)
    {
        //printf("i = %d\n", i);
        j = i;
        while(j)
        {
            j = next[j];
            if(s[i] == s[j])
            {
                next[i+1] = j+1;
                break;
            }
        }
    }
}
char s[maxn];
int main()
{
    while(scanf("%s", s))
    {
        if(s[0] == '.') break;
        int n = strlen(s), prefix = n, ok = 0;
        pre_kmp(s, n);
/*
        while(prefix)
        {
            //printf("prefix\n");
            prefix = next[prefix];
            if(n % (n - prefix) == 0)
            {
                cout << n / (n - prefix) << endl;
                ok = 1;
                break;
            }
        }
        */
        if(n && n % (n - next[prefix]) == 0) cout << n / (n - next[prefix]) << endl;
        else cout << 1 << endl;
        //if(!ok) cout << 1 << endl;
    }
    return 0;
}