acwing KMP算法

【数据结构】KMP算法（详解）

#include <iostream>
#include <string>
#include <vector>

using namespace std;
const int N = 1e6 + 10; 

int n1, n2;
char s1[N], s2[N];
vector<int> next_val(N);

void get_next()
{
    // next数组中的元素为最长比配前缀的下标
    // 这里下标从一开始，无匹配前缀的元素为0
    // 这里的前缀匹配字符为 j 的下一个字符，j从0开始，方便填值
    // 后缀匹配字符从i=2开始，第一个字符的next值默认为0
    for(int i = 2, j = 0; i <= n1; ++i)
    {
        // 前后匹配到不同字符时，j依次向前查找，而不是直接退回起点
        // 因为j前面的next已经明确了，可以对自身进行一次next查询
        // ：j>0保证j位置next为有效下标
        while(j > 0 && s1[j + 1] != s1[i])
            j = next_val[j];
        
        if(s1[j + 1] == s1[i])
            j++; // 匹配一个字符，j 后移
        
        // 每个检测next值的i位置都要填值
        next_val[i] = j; 
    }
    
    // for(int i = 1; i <= n1; i++)
    //     cout << next_val[i] << " ";
}

void match_trig()
{
    // i是S的下标，从1开始，j是P的下标，从0开始，j+1位进行比对
    for(int j = 0, i = 1; i <= n2; ++i)
    {
        // 同上
        while(j > 0 && s1[j + 1] != s2[i])
            j = next_val[j];
        if(s1[j + 1] == s2[i])
            j++;
        
        // j到s1尾部，输出i匹配的起始位置
        if(j == n1)
        {
            cout << i - n1 << " ";
            // j跳到next指向的元素位置
            j = next_val[j];        
        }
    }
}

int main()
{
    cin >> n1 >> s1 + 1 >> n2 >> s2 + 1;
    get_next();
    match_trig();
    return 0;
}