扩展kmp

问题描述
给定串S和T，设长度分别为n,m，依次计算出extend[0~n-1]，extend[i]表示S的后缀S[i,n-1]与T的最长公共前缀。

算法描述

为了计算extend数组，我们先计算一个数组next[0~m-1],next[i]表示T的后缀T[i,m-1]与T[0,m-1]的最长公共前缀。

假设已经计算出了next数组。现在借助于next数组，从前向后计算S的每个位置的extend值。假设当前遍历到位置i，即extend[0~i-1]都已经计算完。对于所有的$jin [0,i-1]$，j对应一个f(j)=j+extend[j]-1，设f(j)的最大值为p，取得p的位置为a，即p=f(a)。

S串： 0 1 2 … a … i-1 i … p …

对于a这个位置，有S[a,p]=T[0,p-a]，那么有S[i,p]=T[i-a,p-a]

要求extend[i]，即S[i,n-1]匹配T[0,m-1]，我们可以先用T[i-a,m-1]来匹配T[0,m-1]，这正是next[i-a]表示的含义：T中位置i-a开始的后缀跟T的最长公共前缀长度。令L=next[i-a]，分以下两种情况讨论：
第一种 $i+L<p$，如下所示。那么有exend[i]=next[i-a](即L)。 

第二种 $i+Lgeq p$，如下所示

p+1及其之后我们不知道是不是匹配，所以要逐个比较。

下面是计算extend的代码。

void GetExtend(
        const char* S,const int nSLen,
        const char* T,const int nTLen,
        const int *Next,
        int *Extend)
{
    int a,p,j=-1;
    for(int i=0;i<nSLen;++i,--j)
    {
        if(j<0||i+Next[i-a]>=p)
        {
            if(j<0) j=0,p=i;
            while(p<nSLen&&j<nTLen&&S[p]==T[j]) ++j,++p;
            Extend[i]=j;
            a=i;
        }
        else Extend[i]=Next[i-a];
    }
}

对于next的求法，显然与上面是一个过程，只是两个串一样：

void GetNext(
        const char* S,const int nSLen,
        int *Next)
{
    int a,p,j=-1;
    Next[0]=nSLen;
    for(int i=1;i<nSLen;++i,--j)
    {
        if(j<0||i+Next[i-a]>=p)
        {
            if(j<0) j=0,p=i;
            while(p<nSLen&&S[p]==S[j]) ++j,++p;
            Next[i]=j;
            a=i;
        }
        else Next[i]=Next[i-a];
    }
}