基本算法之KMP算法

KMP算法

作用

字符串匹配算法，暴力搜索的时间复杂度达到O(n^2)，KMP算法时间复杂度O(m + n),其中n是待查找的字符串长度，m是需要匹配的字符串长度。

代码

#include <bits/stdc++.h>
using namespace std;

class KMP{
private:
	vector<vector<int> > dp;
	string pat;
public:
	KMP(string& pat){
		this->pat = pat;
		int M = pat.size();
		dp = vector<vector<int>> (M,vector<int>(256,0));
		dp[0][pat[0]] = 1;
		int X = 0;
		for (int j = 1;j < M;++j){
			for (int c = 0;c < 256;++c) dp[j][c] = dp[X][c];
			dp[j][pat[j]] = j + 1;
			X = dp[X][pat[j]];
		}

	}
	int search(string& txt){
		int M = pat.size();
		int N = txt.size();
		int j = 0;
		for (int i = 0;i < N;++i){
			j = dp[j][txt[i]];
			if (j == M) return i - M + 1;
		}
		return -1;
	}
};

int main(void){
	string txt = "hello";
	string pat = "ll";
	KMP kmp(pat);
	cout << kmp.search(txt) << endl;
	return 0;
}

进阶

KMP算法适用于单个模式（一个需要匹配的字符串）字符串匹配问题。如果需要多个模式的字符串匹配，可以使用效率更高的字典树结构。