题意:给定若干只含小写字母的字符串(这些字符串总长≤400000),在每个字符串中求出所有既是前缀又是后缀的子串长度。
例如:ababcababababcabab,既是前缀又是后缀的子串:ab,abab,ababcabab,ababcababababcabab。
0 1 2 3 4 5 6 7 8 9 .... 17
分析: S: a b a b c a b a b | a b a b c a b a b len=18
对应的Next数组:0 0 1 2 0 1 2 3 4 | 1 2 3 4 5 6 7 8 9
我们要求的是串S中既是前缀也是后缀的长度,而Next[i]记录的是S[0]~S[i]中
最长前后缀的长度,在这里我们只需要对Next数组递归回去,就可以得到
“所有既是前缀又是后缀的子串长度。” 为什么呢
1. Next[len-1]=9,
2. Next[ Next[len-1]-1 ]=4,
3. Next[ Next[ Next[len-1]-1 ]-1 ]=2,
4. Next[ Next[ Next[ Next[len-1]-1 ]-1 ]-1 ]=0;
1. Next[len-1]=9, 那么S[0...(9-1)]=S[9...17]
2. Next[ Next[len-1]-1 ]=4, 那么S[5...8]=S[0...3],由1我们
知道 S[5...8]=S[14..17],所以S[0...3]=S[14..17]。
3. Next[ Next[ Next[len-1]-1 ]-1 ]=2, 那么S[0...1]=S[2...3],
由2知道,S[2...3]=S[7...8];
由1,S[7...8]=S[16...17];所以S[0...1]=S[16...17].
#include <set>
#include <map>
#include <cmath>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include<cstring>
#include <stdio.h>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn = 1e6+5;
char p[maxn];
int Next[maxn],ans[maxn];
void KMP_Next()
{
memset(Next,0,sizeof(Next));
memset(ans,0,sizeof(ans));
int pos=0,t;
int len=strlen(p);
for(int i=1,j=0;i<len;i++){
while (j&&p[i]!=p[j]) j=Next[j-1];
if(p[i]==p[j]) j++;
Next[i]=j;
}
ans[pos]=len-1;
t=len-1;
while (Next[t]){
ans[++pos]=Next[t-1];
t=Next[t-1];
}
for(int i=pos;i>=0;i--)
printf("%d ",ans[i]+1);
printf("\n");
}
int main()
{
while (~scanf("%s",p)){
KMP_Next();
}
return 0;
}