首页 > 代码库 > POJ2406 KMP算法

POJ2406 KMP算法

POJ2406

问题重述:给定字符串s0,记连续的k个s前后相连组成的s0s0...s0为s0^k。输入字符串S,求最大的k,使得S = s0^k.

问题分析:

1.采用kmp算法求出前缀函数 prefix[i] (i = 1,2,... n, n = length(S)).

2.假如n - prefix[n]能够整除n, 则S能表示为k = n / (n - prefix[n])个连续s0,且k此时取到最大值

3.否则, S不能表示为连续的s0的形式, k = 1。

AC代码:

 1 //Memory: 5680K        Time: 313MS 2 #include <iostream> 3 #include <cstring> 4 #include <string> 5  6 using namespace std; 7  8 const int maxn = 1000010; 9 10 int prefix[maxn];11 string s;12 13 void init()14 {15     int l = s.size();16     memset(prefix, 0, sizeof(prefix));17     int k = 0;18     for (int i = 2; i <= l; i++) {19         while (k > 0 && s[k] != s[i - 1])20             k = prefix[k];21         if (s[k] == s[i - 1])22             k++;23         prefix[i] = k;24     }25 }26 27 int main()28 {29     while (cin >> s && s != ".") {30         init();31         int l = s.size();32         33         if ( prefix[l] != 0 && l % (l - prefix[l]) == 0) 34             cout << l / (l - prefix[l]) << endl;35         else36             cout << "1" << endl;37     }38     return 0;39 }

 

POJ2406 KMP算法