KMP算法

KMP算法模板题

（选自hiho一下 第三周的题目： http://hihocoder.com/contest/hiho3/problem/1）

输入

第一行一个整数N，表示测试数据组数。

接下来的N*2行，每两行表示一个测试数据。在每一个测试数据中，第一行为模式串，由不超过10^4个大写字母组成，第二行为原串，由不超过10^6个大写字母组成。

其中N<=20

输出

对于每一个测试数据，按照它们在输入中出现的顺序输出一行Ans，表示模式串在原串中出现的次数。

样例输入 

5

HA

HAHAHA

WQN

WQN

ADA

ADADADA

BABABB

BABABABABABABABABB

DAD

ADDAADAADDAAADAAD

样例输出

3

1

3

1

0

 

时间限制:1000ms　　　　单点时限:1000ms　　　　　　　　内存限制:256MB

 

 

 提交时，CE 提示说 数组 next[] 是不合法的 = =，硬生生改成 nxt[]

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
using namespace std;

const int maxn = 1e4 + 5;
int nxt[maxn];
string S, T;

void get_next()
{
    int i = 0, j = -1;
    nxt[0] = -1;
    while ( i < T.size()) {
        if (j == -1 || T[i] == T[j]) {  // T[i]:后缀的单个字符； T[j]:前缀的单个字符
            ++i;
            ++j;
            nxt[i] = j;
        }
        else j = nxt[j];
    }
  //  for (i = 0 ; i < T.size(); i++) {
  //      printf("next[%d] = %d
", i, next[i]);
  //  }
}

int kmp()
{
    get_next();
    int i = 0, j = 0, Ans = 0;
    while (i < S.size()) {
        if (j == -1 || S[i] == T[j]) {
            ++i;
            ++j;
        }
        else
            j = nxt[j];       // 指针后退重新开始匹配
        if (j == T.size()) {   // 匹配
            Ans++;
        }
    }
    return Ans;
}

int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt", "r", stdin);
    #endif // ONLINE_JUDGE

    int n;
    while (scanf("%d", &n) !=EOF) {
        while (n--) {
            cin >> T >> S;
            cout << kmp() << endl;
        }
    }
    return 0;
}