POJ2752 Seek the Name, Seek the Fame KMP应用

POJ2752 Seek the Name, Seek the Fame

这算是一道比较脑洞的KMP题目。
先来看题意。对于给定的一个字符串，有一部分前缀和后缀是相同的，让你从小到大输出这些相同前缀后缀的长度。
如样例 ababcababababcabab，相同的前缀后缀有“ab”,”abab”,”ababcabab”,”ababcababababcabab”四个，长度分别为2 4 9 18；

然后来看一看解法：

对于一个字符串 s[0…n-1] ,假设 s[0…i] , s[j…n-1] 是相同的，那么所有比它小的前缀后缀，如 s[0…x] , s[y…n-1] 都可以放在s[0…i]中( s[0…x] , s[z…i] )。那么，我们就可以从n-1开始，从大到小找。

---

代码

#include <iostream>#include <stack>#include <cstring>using namespace std;char s[400005];int next[400005];stack <int> st;void get_next(int len){ // 求next数组    int i = 0, j = -1;    next[0] = j;    while(i != len){        if(j == -1 || s[i] == s[j]) next[++i] = ++j;        else j = next[j];    }}int main(){    while(cin >> s){        int l = strlen(s);        get_next(l);        int i = l;        while(i != 0){ // 从长度l开始，直到i == 0为止            st.push(i);//把答案放在栈中，反向输出。            i = next[i];        }        while(!st.empty()){//利用栈的特性，后入先出，即可满足题目要求            cout << st.top()<<' ';st.pop();        }        cout << endl;    }    return 0;}