【poj 3461】Oulipo（字符串--KMP）

题意：求子串在文本串中出现了多少次。

解法：使用KMP的next[ ]和tend[ ]数组计数。

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

const int N=10010,M=1000010;
char s[N],ss[M];
int n,m;
int next[N];//,tend[M];

void kmp()
{
    memset(next,0,sizeof(next));
    int p=0;
    next[1]=0;
    for (int i=2;i<=n;i++)
    {
      while (s[i]!=s[p+1] && p) p=next[p];
      if (s[i]==s[p+1]) p++;
      next[i]=p;
    }
    //memset(tend,0,sizeof(tend));
    int cnt=0; p=0;
    for (int i=1;i<=m;i++)
    {
      while (ss[i]!=s[p+1] && p) p=next[p];
      if (ss[i]==s[p+1]) p++;
      //tend[i]=p;
      if (p==n) cnt++;
    }
    printf("%d
",cnt);
}
int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
      scanf("%s%s",s+1,ss+1);
      n=strlen(s+1),m=strlen(ss+1);
      kmp();
    }
    return 0;
}