模板 - 字符串

要先理解前缀函数的定义，前缀函数 \(\pi(i)\) 表示字符串 \(s[0,i]\) 的同时是其最长真前缀及最长真后缀的长度，简单来说就是这个 \(s[0,i]\) 首尾最长的重叠长度（不能完全重叠）。

注意这里的字符串都是从0开始计数的。

int pi[1005];

void GetPrefixFunction(char *s, int sl) {

    pi[0] = 0, pi[1] = 0;

    for(int i = 1, k = 0; i < sl; ++i) {

        while(k && s[i] != s[k])

            k = pi[k];

        pi[i + 1] = (s[i] == s[k]) ? ++k : 0;

    }

}

//返回t在s中所有occurrence的首地址，s和t都是从0开始的

int ans[1005], atop;

void KMP(char *s, int sl, char *t, int tl) {

    GetPrefixFunction(t, tl);

    atop = 0;

    for(int i = 0, k = 0; i < sl; ++i) {

        while(k && s[i] != t[k])

            k = pi[k];

        k += (s[i] == t[k]);

        if(k == tl)

            ans[++atop] = i - tl + 1;

    }

}

巴特西

模板 - 字符串 - KMP算法

最新文章

热门文章