C++ KMP算法完整实现

#include <iostream>
#include<vector>
#include<string>
#include <unordered_map>
#include <unordered_set>
#include <queue>
#include <algorithm>//算法头文件
#include <numeric>
#include <stack>
#include<typeinfo>

#include<stdio.h>
using namespace std;

class KMP {
private:
    vector<vector<int>> dp;//二维状态转移数组
    string pat;//模式串
public:
     KMP(string _pat) {
         pat = _pat;
         int len = pat.size();

         //状态转移数组
         dp.resize(len, vector<int>(256));//指定大小后 默认值为0
         //边界条件 在0状态时只有遇到了pat[0]才能转到状态1
         dp[0][(int)pat[0]] = 1;
         //初始化影子状态初始值
         int X = 0;
         //构建状态转移图
         for (int j = 1; j < len; j++) {
             for (int c = 0; c < 256; c++) {
                 if(c != (int)pat[j])
                     dp[j][c] = dp[X][c];            //如果遇到的字符c与j状态需要进行推进所匹配的字符不匹配时 就状态重启到影子状态
                 else
                     dp[j][c] = j + 1;    //如果遇到的字符c与j状态需要进行推进所匹配的字符相匹配时 将进行状态推进
                 //也可以对上面的if进行优化如下面注释的两行
                 //dp[j][c] = dp[X][c];            //如果遇到的字符c与j状态需要进行推进所匹配的字符不匹配时 就状态重启到影子状态
                 //dp[j][(int)pat[j]] = j + 1;    //如果遇到的字符c与j状态需要进行推进所匹配的字符相匹配时 将进行状态推进
             }
             //更新影子状态
             X = dp[X][(int)pat[j]];
         }
    }
     int search(string txt) {
         int len = txt.size();

         //pat串的初始状态为0
         int j = 0;
         for (int i = 0; i < len; i++) {
             //计算j的下一个状态
             j = dp[j][(int)txt[i]];
             //如果状态到达终止态(到pat末尾时)
             if (j == pat.size()) return i - pat.size() + 1;
         }
         //如果遍历完txt还没到达终止态则匹配失败
         return -1;
     }
};

int main()
{
    string pat = "aaab";//ababc  aaab
    string txt = "aaaaaaab"; //aaaaaaab  aaacaaab
    KMP aa(pat);
    int pos1 = aa.search(txt);
    cout << pos1 << endl;
    system("pause");
    return 0;
}