拓展kmp是對(duì)KMP算法的擴(kuò)展,它解決如下問(wèn)題:
定義母串S,和字串T,設(shè)S的長(zhǎng)度為n,T的長(zhǎng)度為m,求T與S的每一個(gè)后綴的最長(zhǎng)公共前綴,也就是說(shuō),設(shè)extend數(shù)組,extend[i]表示T與S[i,n-1]的最長(zhǎng)公共前綴,要求出所有extend[i] (0<=i<n)。
注意到,如果有一個(gè)位置extend[i]=m,則表示T在S中出現(xiàn),而且是在位置i出現(xiàn),這就是標(biāo)準(zhǔn)的KMP問(wèn)題,所以說(shuō)拓展kmp是對(duì)KMP算法的擴(kuò)展,所以一般將它稱為擴(kuò)展KMP算法。
參考博客一
參考博客二
然后每次寫模板的時(shí)候默默把這張圖畫出來(lái)就會(huì)寫了
擴(kuò)展KMP_2.png
擴(kuò)展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//第二種情況,要繼續(xù)匹配才能得到next[i]的值
{
j=next[p0]+p0-i;//如果i>po+next[po],則要從頭開(kāi)始匹配
if(j<0) j=0;
while(i+j<len&&str[i+j]==str[j]) j++;//計(jì)算next[i]
next[i]=j;
p0=i;
}
}
}
void exkmp(char *str,char *p)//計(jì)算extend數(shù)組
{
int i=0,j,p0,slen=strlen(str),plen=strlen(p);
getNext(p);//計(jì)算p的next數(shù)組
while(i<slen&&i<plen&&str[i]==p[i]) i++;//計(jì)算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 //第二種情況,要繼續(xù)匹配才能得到ex[i]的值
{
j=extend[p0]+p0-i;
if(j<0) j=0;//如果i>ex[po]+po則要從頭開(kāi)始匹配
while(i+j<slen&&j<plen&&str[i+j]==p[j]) j++;//計(jì)算ex[i]
extend[i]=j;
p0=i;//更新po的位置
}
}
}