POJ2752 Seek the Name,Seek the Fame KMP算法

KMP继续练手.题目问的是一个串前缀等于后缀的可能长度是哪些,输出来.题目考的是对KMP失配指针的理解,当最后一位失配(即'\0'那里)时,指针会移动到前缀对应匹配的部分,所以这个长度是我们要的,然后接着这个新的前缀的失配指针移到的部分,与这个前缀的后缀也是匹配的..这样一直滚下去就可以了得到所有可能的值.

贴一记代码.

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
#define mxp 400050

int f[mxp+50];
char P[mxp+50];

void getFail(const char *P,int *f)
{
	int m=strlen(P);
	f[0]=f[1]=0;
	for(int i=1;i<m;++i){
		int j=f[i];
		while(j&&P[i]!=P[j]) j=f[j];
		f[i+1]= P[i]==P[j]? j+1:0;
	}
}

int main()
{
	while(scanf("%s",P)!=EOF)
	{
		getFail(P,f);
		int n=strlen(P);
		vector<int> ans;
		while(n){
			ans.push_back(n);
			n=f[n];
		}
		reverse(ans.begin(),ans.end());
		printf("%d",ans[0]);
		for(int i=1;i<ans.size();++i){
			printf(" %d",ans[i]);
		}
		puts("");
	}
	return 0;
}