参考链接:扩展KMP
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=9999;
int Next[maxn],extand[maxn];
void getnext(char* t)
{
int i,len=strlen(t);
Next[0]=len;
for(i=0;i<len-1&&t[i]==t[i+1];i++);
Next[1]=i;
int a=1;
for(int k=2;k<len;k++){
int p=a+Next[a]-1,l=Next[k-a];
if(k+l-1>=p){
int j=max(p-k+1,0);
while(k+j<len&&t[k+j]==t[j]) j++;
Next[k]=j;a=k;
}else Next[k]=l;
}
}
void getextend(char* s,char* t)
{
memset(Next,0,sizeof(Next));
getnext(t);
int slen=strlen(s),tlen=strlen(t),a=0;
int minlen=min(slen,tlen);
while(a<minlen&&s[a]==t[a]) a++;
extand[0]=a,a=0;
for(int k=1;k<slen;k++){
int p=a+extand[a]-1,l=Next[k-a];
if(k+l-1>=p){
int j=max(p-k+1,0);
while(k+j<slen&&j<tlen&&s[k+j]==t[j]) j++;
extand[k]=j;a=k;
}else extand[k]=l;
}
}
int main()
{
char s[maxn],t[maxn];
while(scanf("%s%s",s,t)!=EOF){
getextend(s,t);
for(int i=0,len=strlen(t);i<len;i++)
printf("%d ",Next[i]);
puts("");
for(int i=0,len=strlen(s);i<len;i++)
printf("%d ",extand[i]);
puts("");
}
}