1458：Seek the Name, Seek the Fame （KMP）

1458：Seek the Name, Seek the Fame

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 643 通过数: 339
【题目描述】
原题来自：POJ 2752

给定若干字符串（这些字符串总长 ≤4×105​​ ），在每个字符串中求出所有既是前缀又是后缀的子串长度。

例如：ababcababababcabab，既是前缀又是后缀的：ab，abab，ababcabab，ababcababababcabab。

【输入】
输入若干行，每行一个字符串。

【输出】
对于每个字符串，输出一行，包含若干个递增的整数，表示所有既是前缀又是后缀的子串长度。

【输入样例】
ababcababababcabab
aaaaa
【输出样例】
2 4 9 18
1 2 3 4 5

#include <bits/stdc++.h>

using namespace std;
const int N=4e5+5;
string s;
int a[N],res[N],n,m;

void get_next()
{
    int i=0,j=-1;
    a[0]=-1;
    while(i<n)
    {
        if(j==-1||s[i]==s[j])
            i++,j++,a[i]=j;
        else
            j=a[j];
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    while(cin >> s)
    {
        m=0;
        n=s.size();
        get_next();
        res[m++]=n;
        res[m++]=a[n];//长度为n的串的最长公共前后缀长度
        for(int i=a[n];i;i=a[i])//i为长度
            res[m++]=a[i];
        for(int i=m-2;i>=0;i--)
            printf("%d ",res[i]);
        printf("\n");
    }
    return 0;
}