扩展kmp

扩展kmp

扩展kmp处理的问题:

字符串S和字符串T,求S的每个后缀与T的最长公共前缀

nxt数组与kmp的不一样

char S[N], T[N];
int n, m, nxt[N], extend[N];
//nxt[i]表示从T[i]开始的后缀与T的最长公共和前缀
//extend[i]表示从S[i]开始的后缀与T的最长公共和前缀
//n,m表示S,T的长度,字符串从0开始
void GetNext()
{
    int l = 0, r = 0;
    nxt[0] = m;
    for (int i = 1; i < m; i++)
    {
        if (i >= r || i + nxt[i - l] >= r)
        {
            if (i >= r)
                r = i;
            while (r < m && T[r] == T[r - i])
                r++;
            nxt[i] = r - i;
            l = i;
        }
        else
            nxt[i] = nxt[i - l];
    }
}
void GetExtend()
{
    int l = 0, r = 0;
    GetNext();
    for (int i = 0; i < n; i++)
    {
        if (i >= r || i + nxt[i - l] - 1 >= r - 1)
        {
            if (i >= r)
                r = i;
            while (r < n && r - i < m && S[r] == T[r - i])
                r++;
            extend[i] = r - i;
            l = i;
        }
        else
            extend[i] = nxt[i - l];
    }
}