HDU2594 Simpsons’ Hidden Talents 拓展KMP

HDU2594 Simpsons’ Hidden Talents

nxt[ i ]表示s1[ i ~ m - 1 ] 与s1[ 0 ~ m - 1 ]的最长公共前缀

extend[ i ]表示s2[ i ~ n - 1 ] 与s1[ 0 ~ m - 1 ]的最长公共前缀

求最长的  s1的前缀==s2的后缀  即

#include<bits/stdc++.h>
using namespace std;
const int MAX=5*1e4+5;
char s1[MAX],s2[MAX];
int nxt[MAX],ext[MAX],L1,L2;
void build()
{
    memset(nxt,0,sizeof(nxt));
    nxt[0]=L1;
    int i=0;while(i+1<L1&&s1[i]==s1[i+1]) ++i;
    nxt[1]=i;
    int j=1;
    for(int k=2;k<L1;++k)
    {
        int p=nxt[j]+j-1;
        int q=nxt[k-j];
        if(k+q<p+1) nxt[k]=q;
        else
        {
            i=max(p-k+1,0);
            while(i+k<L1&&s1[k+i]==s1[i]) ++i;
            nxt[k]=i;j=k;
        }
    }
}
void extend()
{
    L1=(int)strlen(s1);L2=(int)strlen(s2);
    build();
    memset(ext,0,sizeof(ext));
    int i=0;while(i<L1&&i<L2&&s2[i]==s1[i]) ++i;
    ext[0]=i;
    int j=0;
    for(int k=1;k<L2;++k)
    {
        int p=ext[j]+j-1;
        int q=nxt[k-j];
        if(k+q<=p) ext[k]=q;
        else
        {
            i=max(p-k+1,0);
            while(i+k<L2&&i<L1&&s2[k+i]==s1[i]) ++i;
            ext[k]=i;j=k;
        }
    }
}
int main()
{
    while(~scanf("%s%s",s1,s2))
    {
        extend();
        int ans=0;
        for(int i=0;i<L2;i++) if(ext[i]==L2-i)ans=max(ans,ext[i]);
        if(ans>0) {for(int i=0;i<ans;i++) putchar(s1[i]);putchar(' ');}
        printf("%d\n",ans);
    }
    return 0;
}