KMP笔记

KMP算法

下面是我写的一个kmp查找类

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

class Kmp{
public:
    Kmp(string t) : t(t){}        // 利用模式串进行初始化       
    
    // 找到第一次出现的位置, x ： 查找起始点
    int find(int x, string s) {

        next = GetNext(s);

        int i = 0;
        for(int j = x; j < t.size(); j++) {
            while(s[i] != t[j] && i > 0) {
                i = next[i - 1];
            }

            if(s[i] == t[j]) i++;

            /**
             * @brief Des
             * 此处稍加修改可以完成不同功能
             * 1. if(i == s.size()) i == 0; 再次从头找, 用一个count计数，
             * 实现不重复的查找相同字串的个数
             * 
             * 2. if(i == s.size()) i == next[i - 1]; 再次从最长前缀开始找,
             * 用count计数，实现可重复的查找相同字符串的个数
             */

            if(i == s.size()) return j - i + 1;
        }                                        

        return -1;
    }

    int find(string s) {

        next = GetNext(s);

        int i = 0;
        for(int j = 0; j < t.size(); j++) {
            while(s[i] != t[j] && i > 0) {
                i = next[i - 1];
            }

            if(s[i] == t[j]) i++;

            /**
             * @brief Des
             * 此处稍加修改可以完成不同功能
             * 1. if(i == s.size()) i == 0; 再次从头找, 用一个count计数，
             * 实现不重复的查找相同字串的个数
             * 
             * 2. if(i == s.size()) i == next[i - 1]; 再次从最长前缀开始找,
             * 用count计数，实现可重复的查找相同字符串的个数
             */

            if(i == s.size()) return j - i + 1;
        }                                        

        return -1;
    }
private:
    vector<int> next; // 前缀表
    string t;

    vector<int> GetNext(string s) {
        vector<int> res(s.size());
        res[0] = 0;
        int i = 0;

        for(int j = 1; j < s.size(); j++) {
            while(i > 0 && s[i] != s[j]) {
                i = res[i - 1];
            }

            if(s[i] == s[j]) i++;

            res[j] = i;
        }

        return res;
    } 

};

int main() {
    string s = "aabaabaabaaf";
    //          012345678901
    Kmp kmp(s);
    cout<<kmp.find("aabaaf")<<endl;
}