KMP 最模板

转

模板如下：

#include <iostream>  
#include <cstring>  
using namespace std;  
  
const int N = 1000002;  
int next[N];  
char S[N], T[N];  
int slen, tlen;  
  
void getNext()  
{  
    int j, k;  
    j = 0; k = -1; next[0] = -1;  
    while(j < tlen)  
        if(k == -1 || T[j] == T[k])  
            next[++j] = ++k;  
        else  
            k = next[k];  
  
}  
/* 
返回模式串T在主串S中首次出现的位置 
返回的位置是从0开始的。 
*/  
int KMP_Index()  
{  
    int i = 0, j = 0;  
    getNext();  
  
    while(i < slen && j < tlen)  
    {  
        if(j == -1 || S[i] == T[j])  
        {  
            i++; j++;  
        }  
        else  
            j = next[j];  
    }  
    if(j == tlen)  
        return i - tlen;  
    else  
        return -1;  
}  
/* 
返回模式串在主串S中出现的次数 
*/  
int KMP_Count()  
{  
    int ans = 0;  
    int i, j = 0;  
  
    if(slen == 1 && tlen == 1)  
    {  
        if(S[0] == T[0])  
            return 1;  
        else  
            return 0;  
    }  
    getNext();  
    for(i = 0; i < slen; i++)  
    {  
        while(j > 0 && S[i] != T[j])  
            j = next[j];  
        if(S[i] == T[j])  
            j++;  
        if(j == tlen)  
        {  
            ans++;  
            j = next[j];  
        }  
    }  
    return ans;  
}  
int main()  
{  
      
    int TT;  
    int i, cc;  
    cin>>TT;  
    while(TT--)  
    {  
        cin>>S>>T;  
        slen = strlen(S);  
        tlen = strlen(T);  
        cout<<"模式串T在主串S中首次出现的位置是: "<<KMP_Index()<<endl;  
        cout<<"模式串T在主串S中出现的次数为: "<<KMP_Count()<<endl;  
    }  
    return 0;  
}  

看上串在下串中出现的次数 

3
BAPC
BAPC
AZA
AZAZAZA
VERDI
AVERDXIVYERDIAN

#include <iostream>
#include <cstring>
#include<stdio.h>
using namespace std;


const int N = 1000002;
int next[N];
char S[N], T[N];
int slen, tlen;


void getNext()
{
    int j, k;
    j = 0; k = -1; next[0] = -1;
    while(j < tlen)
        if(k == -1 || T[j] == T[k])
            next[++j] = ++k;
        else
            k = next[k];


}
/*
返回模式串T在主串S中首次出现的位置
返回的位置是从0开始的。
*/
int KMP_Index()
{
    int i = 0, j = 0;
    getNext();


    while(i < slen && j < tlen)
    {
        if(j == -1 || S[i] == T[j])
        {
            i++; j++;
        }
        else
            j = next[j];
    }
    if(j == tlen)
        return i - tlen;
    else
        return -1;
}
/*
返回模式串在主串S中出现的次数
*/
int KMP_Count()
{
    int ans = 0;
    int i, j = 0;
    if(slen == 1 && tlen == 1)
    {
        if(S[0] == T[0])
            return 1;
        else
            return 0;
    }
    getNext();
    for(i = 0; i < slen; i++)
    {
        while(j > 0 && S[i] != T[j])
            j = next[j];
        if(S[i] == T[j])
            j++;
        if(j == tlen)
        {
            ans++;
            j = next[j];
        }
    }
    return ans;
}
int main()
{


    int TT;
    int i, cc;
    scanf("%d",&TT);
    while(TT--)
    {
        scanf("%s%s",&T,&S);
        slen = strlen(S);
        tlen = strlen(T);


        printf("%d\n",KMP_Count());
    }
    return 0;
}