HDU - 2594 擴展KMP：s1的前綴和s2的後綴的最長公共長度

題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=2594
題目大意：就是求s1的前綴和s2的後綴的最長公共長度。

擴展KMP的extend[i]含義：

定義母串S，和字串T，設S的長度爲n，T的長度爲m，求T與S的每一個後綴的最長公共前綴，也就是說，設extend數組,

extend[i]表示T與S[ i,n-1 ]的最長公共前綴，

Next[i]表示的是，以i爲開頭的後綴，與整個字符串的最長公共前綴。

那麼因爲題目要求s必須是後綴，那麼必須滿足：extend[i]+i==n

#include<bits/stdc++.h>
using namespace std;

char s[50005], t[50005];
int Next[50005],extend[50005];

//預處理計算Next數組
void getNext(char str[])
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len; //初始化Next[0]
    while(str[i]==str[i+1] && i+1<len) i++; Next[1]=i; //計算Next[1]
    po=1; //初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i < Next[po]+po) //第一種情況，可以直接得到Next[i]的值
            Next[i]=Next[i-po];
        else //第二種情況，要繼續匹配才能得到Next[i]的值
        {
            j = Next[po]+po-i;
            if(j<0) j=0; //如果i>po+Next[po],則要從頭開始匹配
            while(i+j<len && str[j]==str[j+i]) j++; Next[i]=j;
            po=i; //更新po的位置
        }
    }
}

//計算extend數組
void EXKMP(char s1[],char s2[])
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    getNext(s2); //計算子串的next數組
    while(s1[i]==s2[i] && i<l2 && i<len) i++; extend[0]=i;
    po=0; //初始化po的位置
    for(i=1;i<len;i++)
    {
        if(Next[i-po]+i < extend[po]+po) //第一種情況，直接可以得到extend[i]的值
            extend[i]=Next[i-po];
        else //第二種情況，要繼續匹配才能得到extend[i]的值
        {
            j = extend[po]+po-i;
            if(j<0) j=0; //如果i>extend[po]+po則要從頭開始匹配
            while(i+j<len && j<l2 && s1[j+i]==s2[j]) j++; extend[i]=j;
            po=i; //更新po的位置
        }
    }
}

int main()
{
    while(~scanf("%s%s", s, t)){
        EXKMP(t, s);
        int f=0;
        int n=strlen(t), m=strlen(s);
        for(int i=0; i<n-1; i++){
            if(extend[i]+i==n){

                for(int k=i; k<i+extend[i]; k++){
                    printf("%c", t[k]);
                }
                f=1;
                printf(" %d\n", extend[i]);
                break;
            }
        }
        if(f==0){
            printf("0\n");
        }

    }

    return 0;
}