题目链接: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;
}