擴展KMP(模板)

拓展kmp是對KMP算法的擴展,它解決如下問題:
定義母串S,和字串T,設S的長度爲n,T的長度爲m,求T與S的每一個後綴的最長公共前綴,也就是說,設extend數組,extend[i]表示T與S[i,n-1]的最長公共前綴,要求出所有extend[i] (0<=i<n)。
注意到,如果有一個位置extend[i]=m,則表示T在S中出現,而且是在位置i出現,這就是標準的KMP問題,所以說拓展kmp是對KMP算法的擴展,所以一般將它稱爲擴展KMP算法。

const int MAXN=100000;
int next[MAXN];
int extend[MAXN];
void getNext(char *str)
{
    int len=strlen(str),p0,i=0,j;
    next[0]=len;//初始化next[0]
    while(str[i]==str[i+1]&&i+1<len) i++;
    next[1]=i;
    p0=1;//初始化p0
    for(i=2;i<len;i++)
    {
        if(next[i-p0]+i<next[p0]+p0) next[i]=next[i-p0];//第一種情況,可以直接得到next[i]的值
        else//第二種情況,要繼續匹配才能得到next[i]的值  
        {
            j=next[p0]+p0-i;//如果i>po+next[po],則要從頭開始匹配
            if(j<0) j=0;
            while(i+j<len&&str[i+j]==str[j]) j++;//計算next[i]
            next[i]=j;
            p0=i;
        }
    }
}
void exkmp(char *str,char *p)//計算extend數組 
{
    int i=0,j,p0,slen=strlen(str),plen=strlen(p);
    getNext(p);//計算p的next數組
    while(i<slen&&i<plen&&str[i]==p[i]) i++;//計算ex[0]  
    extend[0]=i;
    p0=0;//初始化po的位置  
    for(i=1;i<slen;i++)
    {
        if(next[i-p0]+i<extend[p0]+p0) extend[i]=next[i-p0];//第一種情況,直接可以得到ex[i]的值
        else //第二種情況,要繼續匹配才能得到ex[i]的值  
        {
            j=extend[p0]+p0-i;
            if(j<0) j=0;//如果i>ex[po]+po則要從頭開始匹配
            while(i+j<slen&&j<plen&&str[i+j]==p[j]) j++;//計算ex[i]
            extend[i]=j;
            p0=i;//更新po的位置  
        }
    }
}``
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章