【poj 3461 Oulipo】（KMP模板）

题意：求一个模式串在主串中的出现次数。
思路：KMP

代码：

//#include <bits/stdc++.h>
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

#define mem(a,n) memset(a,n,sizeof(a))
#define rep(i,a,n) for(int i=a;i<n;i++)
#define fre freopen("in.txt","r",stdin)
typedef long long ll;
const ll INF=0x3f3f3f3f;
const int N=1e7+5;

int nt[10000+5];
char a[N],b[10005];
//参数为模板串和next数组
//字符串均从下标0开始
void kmp_next(char *T,int *nt) {
    nt[0]=0;
    for(int i=1,j=0,m=strlen(T); i<m; i++) {
        while(j&&T[i]!=T[j])
            j=nt[j-1];
        if(T[i]==T[j])
            j++;
        nt[i]=j;
    }
}
int kmp(char *S,char *T,int *nt) {
    kmp_next(T,nt);
    int ans=0,sn=strlen(S),tn=strlen(T);
    for(int i=0,j=0; i<sn; i++) {
        while(j&&S[i]!=T[j])
            j=nt[j-1];
        if(S[i]==T[j])
            j++;
        if(j==tn)
            ans++;
    }
    return ans;
}
int main() {
    int T;
    scanf("%d",&T);
    while(T--) {
        scanf("%s%s",b,a);
        printf("%d\n",kmp(a,b,nt));
    }
    return 0;
}