AcWing 831. KMP字符串

#include <iostream>
using namespace std;
const int N = , M = ;
int n, m;
int ne[N];//ne[i] : 以i为结尾的部分匹配的值
char s[M], p[N];
int main() {
    cin >> n >> p +  >> m >> s + ;
    //求next过程
    //i从2开始，因为next[1]=0
    //如果第一个字母失败了，只能从0开始
    for (int i = , j = ; i <= n; i ++ ) {
        while (j && p[i] != p[j + ])    j = ne[j];
        if (p[i] == p[j + ])    j ++ ;
        ne[i] = j;
    }
    //kmp匹配过程
    for (int i = , j = ; i <= m; i ++ ) {
        while (j && s[i] != p[j + ])    //当j没有退回起点，并且当前的s[i]不能和下一个j的位置匹配
            j = ne[j];//移动，保证之前的相等  直到匹配位置，或者j已经到开头了
        if (s[i] == p[j + ]) //如果已经匹配了
            j ++ ;  //j往下移动
        if (j == n) {//说明匹配成果
            printf("%d ", i - n);
            j = ne[j];
        }
    }
    return ;
}

\