1. 程式人生 > >不會打系列-KMP封裝好的模板

不會打系列-KMP封裝好的模板

const int N = 100000;
/*
    KMP 演算法模板
    時間複雜度 : O(n+m) [其中 n,m 是兩字串的長度]
    說明:
        1. a是長串 s是短串 即 a.length >= s.length
        2.實現了“查詢第一次匹配的位置” “匹配的次數” “記錄所有匹配區間”三個功能
        3.三個功能的實現程式碼非常相似,第3個功能可以同時實現前2個的功能
*/
struct KMP
{
    int f[N + 7];
    void getfill (string s) //預處理
    {
        memset (f, 0, sizeof (f) ); //根據其前一個字母得到
        for (int i = 1; i < s.length(); i++)
        {
            int j = f[i];
            while (j && s[i] != s[j])
                j = f[j];
            f[i + 1] = (s[i] == s[j]) ? j + 1 : 0;
        }
    }
    int finds (string a, string s) //找到第一次匹配的位置 a.length >= s.length
    {

        getfill (s);
        int j = 0;
        for (int i = 0; i < a.length(); i++)
        {
            while (j && a[i] != s[j]) j = f[j];
            if (a[i] == s[j]) j++;
            if (j == s.length() )
            {
                return i - s.length() + 1; //返回小串在大串中第一次出現的位置
            }
        }
        return -1;//不匹配返回-1
    }
    int findtimes(string a,string s)//返回匹配次數
    {
        getfill(s);
        int j = 0;int times = 0;
        for (int i = 0;i < a.length();++i)
        {
            while (j&&a[i] != s[j]) j = f[j];
            if (a[i] ==s[j]) j++;
            if (j == s.length())
            {
                times++;
            }
        }
        return times;//不匹配的times將是0
    }
    struct Interval
    {
        int l,r;
    }q[N];//記錄所有匹配區間 [指在長串中的下標,下標從0開始]
    int t;//記錄匹配次數
    void findinterval(string a,string s)//找到並記錄所有的匹配區間
    {
        getfill(s);
        int j = 0;t = 0;
        for (int i = 0;i < a.length();++i)
        {
            while (j&&a[i]!=s[j]) j = f[j];
            if (a[i] == s[j]) j++;
            if (j == s.length())
            {
                t++;
                q[t].l = i - s.length() + 1;
                q[t].r = i;
            }
        }
    }
    void Print()//第3種功能附帶列印功能
    {
        cout<<"兩串匹配了 "<< t <<"次,分別是 :"<<endl;
        for (int i = 1;i <= t;++i)
        {
            cout<<"第"<<i<<"次 : ["<<q[i].l<<","<<q[i].r<<"]\n";
        }
    }
}kmp;