算法基础系列——字符串（二）前缀函数

1、什么是前缀函数

我们通常用$\pi$表示前缀函数

2、代码实现

2.1、朴素写法

使得串下式中的字符串相同，j就是next数组的值。

$$s[0,j]=s[i-j+1,i]$$

// C++ Version
vector<int> prefix_function(string s) {
  int n = (int)s.length();
  vector<int> pi(n);
  for (int i = 1; i < n; i++)
    for (int j = i; j >= 0; j--)
      if (s.substr(0, j) == s.substr(i - j + 1, j)) {
        pi[i] = j;
        break;
      }
  return pi;
}

2.2、优化写法

可以认为优化的写法就是尽可能地保留已经匹配好的前缀，所以可以看成递归的过程，失配的时候不断找满足条件第二长度的位置可以得到状态转移方程
$$j=\pi [j-1]$$

完整推导过程

#include <iostream>
#include <string>
#include <vector>
using namespace std;

vector<int> prefix_function(string s) {
    int n = (int)s.length();
    vector<int> pi(n);
    for (int i = 1; i < n; i++) {
        int j = pi[i - 1];
        while (j > 0 && s[i] != s[j]) j = pi[j - 1]; 

        if (s[i] == s[j]) j++;
        pi[i] = j;
    }
    return pi;
}

int main()
{
    string s = "abcabcd";
    vector<int> a = prefix_function(s);
    for (auto it = a.begin(); it != a.end(); it++)
        cout << *it << " ";
    return 0;
}
//输出：0 0 0 1 2 3 0