首页 > 代码库 > KMP模板

KMP模板

int s[maxn];//文本串
int p[10007];//匹配串
int next[10007];//匹配串的next数组

//优化过后的next 数组求法
void GetNextval(int n)//n代表p数组的长度
{
    int pLen = n;
    next[0] = -1;
    int k = -1;
    int j = 0;
    while (j < pLen - 1)
    {
        //p[k]表示前缀,p[j]表示后缀
        if (k == -1 || p[j] == p[k])
        {
            ++j;
            ++k;
            //较之前next数组求法,改动在下面4行
            if (p[j] != p[k])
                next[j] = k;   //之前只有这一行
            else
                //因为不能出现p[j] = p[ next[j ]],所以当出现时需要继续递归,k = next[k] = next[next[k]]
                next[j] = next[k];
        }
        else
        {
            k = next[k];
        }
    }
}

int KmpSearch(int n,int m)//n,m分别代表文本串和匹配串的长度
{
    int i = 0;
    int j = 0;
    int sLen = n;
    int pLen = m;
    while (i < sLen && j < pLen)
    {
        //①如果j = -1,或者当前字符匹配成功(即S[i] == P[j]),都令i++,j++
        if (j == -1 || s[i] == p[j])
        {
            i++;
            j++;
        }
        else
        {
            //②如果j != -1,且当前字符匹配失败(即S[i] != P[j]),则令 i 不变,j = next[j]
            //next[j]即为j所对应的next值
            j = next[j];
        }
    }
    if (j == pLen)
        return i - j + 1;//返回匹配的开始位置,如果题目下标从1开始则加1,从零开始就不需要
    else
        return -1;
}



模板题:hdu1711

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1e6+7;

int s[maxn];//文本串
int p[10007];//匹配串
int next[10007];//匹配串的next数组

//优化过后的next 数组求法
void GetNextval(int n)//n代表p数组的长度
{
    int pLen = n;
    next[0] = -1;
    int k = -1;
    int j = 0;
    while (j < pLen - 1)
    {
        //p[k]表示前缀,p[j]表示后缀
        if (k == -1 || p[j] == p[k])
        {
            ++j;
            ++k;
            //较之前next数组求法,改动在下面4行
            if (p[j] != p[k])
                next[j] = k;   //之前只有这一行
            else
                //因为不能出现p[j] = p[ next[j ]],所以当出现时需要继续递归,k = next[k] = next[next[k]]
                next[j] = next[k];
        }
        else
        {
            k = next[k];
        }
    }
}

int KmpSearch(int n,int m)//n,m分别代表文本串和匹配串的长度
{
    int i = 0;
    int j = 0;
    int sLen = n;
    int pLen = m;
    while (i < sLen && j < pLen)
    {
        //①如果j = -1,或者当前字符匹配成功(即S[i] == P[j]),都令i++,j++
        if (j == -1 || s[i] == p[j])
        {
            i++;
            j++;
        }
        else
        {
            //②如果j != -1,且当前字符匹配失败(即S[i] != P[j]),则令 i 不变,j = next[j]
            //next[j]即为j所对应的next值
            j = next[j];
        }
    }
    if (j == pLen)
        return i - j + 1;//如果题目下标从1开始则加1,从零开始就不需要
    else
        return -1;
}

int main()
{
    #ifdef xxz
    freopen("in.txt","r",stdin);
    #endif // xxz
    
    int Case;
    cin>>Case;
    while(Case--)
    {
        int n,m;
        cin>>n>>m;
        for(int i = 0; i < n; i++) cin>>s[i];
        for(int i = 0; i < m; i++) cin>>p[i];

        GetNextval( m);
        cout<<KmpSearch( n, m)<<endl;

    }
    return 0;
}


KMP模板