POJ 2752 既是前缀又是后缀的子串 (kmp)

题目链接:POJ 2752


给定一个字符串,求出所有既是前缀又是后缀的子串。


思路

KMP 的next数组的运用(改进前),如果next数组中next[ i ]= k 正值,那么就代表这从 “0 ~ i-1”位置的子串的前缀后缀相同的最大长度为k,并且k代表着与当前后缀最大匹配的前缀位置。


  0 1 2 3 4 5 6 7 8 9 10 11 12
字符串 a a b a a b a a b a a b  
next [ ] -1 0 1 0 1 2 3 4 5 6 7 8 9


如表 next[12 ] = 9 , 9 所在位置的前缀 为 a a b a a b a a b,与后缀是最大匹配 , next[ 9 ] = 6 ,6 所在位置的前缀 为 a a b a a b,仍然和后缀匹配,第二大匹配

next[ 6 ] = 3 , ,a a b, next[ 3 ] = 0,表明需要回溯到头再进行匹配。

因而通过前后缀重复次数的计数,以及next数组的调转,我们就能尽可能少的重复 重复位置的匹配。每次都能跳转到 前后缀最大公共的位置。


因而对于这个题,我们从最后的next[ len ]开始,不断地找 next[next[i]], 所找的前缀都和最后的后缀所匹配,所在位置的值就是前后缀相同的长度。全部累加就可以了。

另外还要加上满足条件的字符串本身。


源代码

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn = 400000 + 10;
const int maxe = 1000000 + 10;
char s1[maxn];
int len;
int nextv[maxn];
int ans[maxn];
void getnext(){
	int i = 0,j = -1;
	nextv[0] = -1;
	while(i<len){
		if(j == -1 || s1[j] == s1[i]){
			nextv[++i] = ++j;
		}
		else
			j = nextv[j];
	}
}
int main(){
	int n;
	while(scanf("%s",s1)!=EOF){
		// memset(nextv,0,sizeof(nextv));
		len = strlen(s1);
		getnext();
		int tmp = len;
		ans[0] = len;
		int cnt = 1;
		while(nextv[tmp]!=0){ 
			ans[cnt++]=nextv[tmp];
			tmp = nextv[tmp];
		}
		for(int i=cnt-1;i>0;i--)
			printf("%d ",ans[i]);
		printf("%d\n",ans[0]);
	}
	return 0;
}


posted @ 2015-10-20 12:31  编程菌  阅读(480)  评论(0编辑  收藏  举报