HDU1686 Oulipo(扩展KMP)

给出单词和文本，求单词在文本中出现了多少次。

求出文本串的ex值，ex[i]表示以i开头，能匹配的模式串的长度。然后统计ex[i]等于模式串长度的有多少个。

next[i]表示以str[i]开头，能与str[0]开始匹配的最大长度。

#include<cstring>
#include<cstdio>
#include<iostream>
#define MAXN 1000010
using namespace std;
char a[MAXN],b[MAXN];
int cnt,lena,lenb,Next[MAXN],ex[MAXN];
void get_next()
{
	memset(Next,0,sizeof Next);
	Next[0] = lena;
	for(int i = 1; a[i] == a[i-1]; i++) Next[1] = i;
	int maxp = Next[1]+1,j = 1,k;
	for(int i = 2; i < lena; i++)
	{
		if(i+Next[i-j] < maxp) Next[i] = Next[i-j];
		else
		{
			for(k = max(i,maxp); k < lenb&&a[k] == a[k-i]; k++);
			j = i;
			Next[i] = k-i>0?k-i:0;
			maxp = Next[i]+i;
		}
	}
}
void get_ex()
{
	int i,j,k;
	memset(ex,0,sizeof ex);
	for(i = 0,j = 0; i < lenb&&j < lena&&b[i] == a[j]; i++,j++);
	ex[0] = j;
	for(i = 1,j = 0; i < lenb&&j < lena&&b[i] == a[j]; i++,j++);
	ex[1] = j;
	int maxp = j+1;
	j = 1;
	for(i = 2; i < lenb; i++)
	{
		if(i+Next[i-j] < maxp) ex[i] = Next[i-j];
		else
		{
			for(k = max(i,maxp); k < lenb&&k-i < lena&&b[k] == a[k-i]; k++);
			j = i;
			ex[i] = k-i>0?k-i:0;
			maxp = k;
		}
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s%s",a,b);
		lena = strlen(a);
		lenb = strlen(b);
		get_next();
		get_ex();
		cnt = 0;
		for(int i = 0; i < lenb; i++)
			if(ex[i] == lena) cnt++;
		printf("%d\n",cnt);
	}
}