扩展KMP算法（详解+代码）

参考链接：扩展KMP

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int maxn=9999;

int Next[maxn];
int extand[maxn];       //s的后缀与t的最长公共前缀。

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){       //l>=p-k+1
            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("");
    }
}