【KMP】ZOJ-3587-Marlon's String

KMP的巧妙应用……

题目

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 100005
int next[N];
long long f[2][N];
void getnext(char str[])
{
	int i,j,len;
	len=strlen(str);
	i=0,j=-1;
	next[0]=-1;
	while(i<len)
	{
		if(j==-1||str[i]==str[j])next[++i]=++j;
		else j=next[j];
	}
}
void KMP(char str[],char st[],int v)
{
	int i,j,len1,len2;
	len1=strlen(str);
	len2=strlen(st);
	i=j=0;
	while(i<len1)
	{
		if(j==-1||str[i]==st[j])
		{
			i++,j++;
			f[v][j]++;
		}
		else j=next[j];
	}
	for(i=len2;i>=0;i--)
		if(f[v][i]>0&&next[i]>=0)
			f[v][next[i]]+=f[v][i];
}
int main()
{
	//freopen("a.txt","r",stdin);
	int i,t,len1,len2;
	long long ans;
	char str0[N],str1[N],st0[N],st1[N];
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s%s",str0,st0);
		len1=strlen(str0);
		len2=strlen(st0);
		for(i=0;i<len1;i++)str1[len1-1-i]=str0[i];
		str1[i]='\0';
		for(i=0;i<len2;i++)st1[len2-1-i]=st0[i];
		st1[i]='\0';
		memset(f,0,sizeof(f));
		getnext(st0);
		KMP(str0,st0,0);
		getnext(st1);
		KMP(str1,st1,1);
		ans=0;
		for(i=1;i<len2;i++)ans+=f[0][i]*f[1][len2-i];
		printf("%lld\n",ans);
	}
	return 0;
}