1、什么是前缀函数
我们通常用
π
π
π表示前缀函数
2、代码实现
2.1、朴素写法
使得串下式中的字符串相同,j就是next数组的值。
s [ 0 , j ] = s [ i − j + 1 , i ] s[0,j]=s[i-j+1,i] 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
=
π
[
j
−
1
]
j=π[j-1]
j=π[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