KMP模板

選自Mr.kuang http://www.cnblogs.com/kuangbin/archive/2012/08/14/2638803.html

/*
* pku3461(Oulipo), hdu1711(Number Sequence)
* 這個模板 字元串是從0開始的
* Next數組是從1開始的
*/
#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;
}
/*
* test case
* aaaaaa a
* abcd d
* aabaa b
*/