poj 2752 求一个字符串所有的相同前后缀

 

求一个字符串所有的相同前后缀
Sample Input

ababcababababcabab
aaaaa
Sample Output

2 4 9 18
1 2 3 4 5

 

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

const int N = 400010;
int next[N];
char  T[N];
int  tlen;
stack<int> st ;

void getNext()
{
    int j, k;
    j = 0; k = -1; next[0] = -1;
    while(j < tlen)
        if(k == -1 || T[j] == T[k])
            next[++j] = ++k;
        else
            k = next[k];

}


int main()
{
    while (cin>>T)
    {
        tlen = strlen(T) ;
        getNext() ;
        int t = next[tlen] ;
        while (t)
        {
            st.push(t) ;
            t = next[t] ;
        }
        while(!st.empty())
        {
            int u = st.top() ;
            printf("%d " , u) ;
            st.pop() ;
        }
        printf("%d\n" , tlen) ;
    }



    return 0;
}