poj 2752 Seek the Name, Seek the Fame

上一篇博客中,可以知道next可以求循环节,于是,这道题就是找前缀和后缀相同。首先自己本身一定是答案,然后,next里面储存着当前已经和前缀匹配的位数,所以直接递归输出next就可以啦。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
char a[400010];
int next[400010];
void print(int who)
{
    if(next[who]==0)
    {
        return ;
    }
    print(next[who]);
    cout<<next[who]<<" ";
}
int main()
{
    while(scanf("%s",a)==1)
    {
        int len=strlen(a);
        next[0]=-1;
        int i=0;
        int j=next[0];
        while(i<len)
        {
            if(j==-1||a[i]==a[j])
            {
                i++;
                j++;
                next[i]=j;
            }
            else
            {
                j=next[j];
            }
        }
        print(len);
        cout<<len<<endl;
    }
    return 0;
}

你可能感兴趣的:(KMP,poj)