zoj 3587 蛮好的KMP题

我以前做过的KMP http://www.cnblogs.com/wuyiqi/tag/KMP/

题意：从S串中拿出两个子串能拼成T串的方案有多少种，两个子串可以重复

这道题其实要解决的问题就是求T串的某个前缀在S串中出现了几次(cnt[])    以及T串的某个后缀在S串中出现了几次(cnt2[])，那么答案就是cnt[i]*cnt2[len-i]的和

具体求的过程就很简单了，cnt[next[i]]+=cnt[i]即可，最后的cnt[i]就表示了i前缀出现的次数，后缀的话将字符串反转再KMP即可

#include
#include
#include
using namespace std;
typedef long long lld;
const int maxn = 100001;
int n,m;
char a[maxn],b[maxn];
int p[maxn],cnt[maxn],cnt2[maxn];
void getp(){
    p[1]=0;
    int i,j=0;
    for(i=2;i<=m;i++){
        while(j>0&&b[j+1]!=b[i]) j=p[j];
        if(b[j+1]==b[i]) j+=1;
        p[i]=j;
    }
}
void kmp(int cnt[])
{
    int i,j=0;
    for(i=1;i<=n;i++){
        while(j>0&&b[j+1]!=a[i]) j=p[j];
        if(b[j+1]==a[i]) j+=1,cnt[j]++;
        if(j==m)    j=p[j];
    }
	for(i=m;i>=1;i--){
		if(p[i]!=0) cnt[p[i]]+=cnt[i];
	}
}
void RE(char s[]){
	int len=strlen(s+1);
	for(int i=1;i<=len/2;i++){
		swap(s[i],s[len-i+1]);
	}
}
lld solve()
{
	lld ans=0;
	for(int i=1;i<=m;i++)
		ans+=(lld)cnt[i]*cnt2[m-i];
	return ans;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
    {
		scanf("%s",a+1);
		scanf("%s",b+1);
        m=strlen(b+1);
        n=strlen(a+1);

		memset(cnt,0,sizeof(cnt));
        getp();
        kmp(cnt);

		RE(a);
		RE(b);
		memset(cnt2,0,sizeof(cnt2));
		getp();
		kmp(cnt2);
		printf("%lld\n",solve());
    }
}