KMP模板(复杂度O(m+n))(新)

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000002;
int nxt[maxn],slen,tlen;
char S[maxn],T[maxn];
void getNxt()
{
    int i,j;
    i=0,j=-1,nxt[0]=-1;
    while(i<tlen)
    {
        if(j==-1||T[i]==T[j])
        {
            nxt[++i]=++j;
        }
        else
        {
            j=nxt[j];
        }
    }
}

int KMP_Index()
{
    int i=0,j=0;
    getNxt();
    while(i<slen&&j<tlen)
    {
        if(j==-1||S[i]==T[j])
        {
            i++;
            j++;
        }
        else
        {
            j=nxt[j];
        }
    }
    if(j==tlen)
    {
        return i-tlen;
    }
    else
        return -1;
}

int KMP_Count()
{
    int ans=0,i=0,j=0;
    if(slen==1&&tlen==1)
    {
        if(S[0]==T[0])
            return 1;
        else
            return 0;
    }
    getNxt();
    for(i=0;i<slen;i++)
    {
        while(j>0&&S[i]!=T[j])
        {
            j=nxt[j];
        }
        if(S[i]==T[j])
            j++;
        if(j==tlen)
        {
            ans++;
            j=nxt[j];//求可重叠子串个数
            //j=0 求不可重叠子串个数
        }
    }
    return ans;
}
int main()
{
    cin >> S >> T;
    slen=strlen(S);
    tlen=strlen(T);
    cout << KMP_Index()<<endl;
    cout << KMP_Count()<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/w1304636468/article/details/89450725
今日推荐