【数据结构】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值默认为0for(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;
}