KMP算法与前缀函数

vector<int> pi(N);
void prefix_function(string s) {//前缀函数
  int n = (int)s.length();
  for (int i = 1; i < n; i++) {
    int j = pi[i - 1];
    while (j > 0 && s[i] != s[j]) j = pi[j - 1];//失配递推找下一个j
    if (s[i] == s[j]) j++;//匹配成功
    pi[i] = j;
  }
}
//默认输入的字符串自s[0]到s[n-1] 
//前缀函数处理 abcabcd
//a[0] 时只有一个元素 默认将其赋值为 0；
//ab=0 abca=1 abcab=2 abcabc=3 abcabcd=0  

vector<int> v;
void find_occurrences(string text, string pattern) {//kmp 函数
  string cur = pattern + '#' + text;
  int sz1 = text.size(), sz2 = pattern.size();
  v.clear();
  prefix_function(cur);
  for (int i = sz2 + 1; i <= sz1 + sz2; i++) {//第cur[sz2]为“*”号
    if (pi[i] == sz2) v.push_back(i - 2 * sz2);
  }
}
//默认输入的字符串自text与pattern自[0]到[n-1]
//输出的v[i]为出现子串的头位置
//abc#abcabc
//v={0,3}