AcWing 831.KMP字符串

AcWing 831.KMP字符串

#include <iostream>

using namespace std;

const int N = 100010, M = 1000010;
// 题意：模式串 P在字符串 S中多次作为子串出现。
// 求出模式串 P在字符串 S中所有出现的位置的起始下标。
int n, m;
int ne[N];
char s[M], p[N];

int main()
{
    // p+1和s+1的目的是 使得字符串从1开始，避免0.
    cin >> n >> p + 1 >> m >> s + 1;

    // 求next数组。在p字符串本身内，找到最长的前后缀，方便ne的指向。
    for (int i = 2, j = 0; i <= n; i++)
    {
        while (j && p[i] != p[j + 1])
            j = ne[j];
        // 如果前后缀相同，此时j++，也就是前后缀长度++
        if (p[i] == p[j + 1])
            j++;
        // 把这次求出的前后缀长度记入ne数组。
        ne[i] = j;
    }
    // kmp算法过程
    for (int i = 1, j = 0; i <= m; i++)
    {
        // 在j可以向后退的前提下，并且s和p的这一位不匹配，j向前退。
        while (j && s[i] != p[j + 1])
            j = ne[j];
        // 如果s和p的这一位匹配成功，那么继续检查下一位。
        if (s[i] == p[j + 1])
            j++;
        // 如果匹配完成，在字符串S中找到了P这一子字符串，输出起始位置。
        // 并且将j向后退，继续检查字符串S后面有没有再次出现子字符串P。
        if (j == n)
        {
            printf("%d ", i - n);
            j = ne[j];
        }
    }

    return 0;
}

PS：看不懂听不懂，自己手写一遍代码的过程就能懂了。