模板 - 字符串 - KMP算法

要先理解前缀函数的定义，前缀函数 \(\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;
    }
}