zoj 3587 蛮好的KMP题

最新推荐文章于 2018-11-11 22:52:50 发布

haha593572013

最新推荐文章于 2018-11-11 22:52:50 发布

阅读量1.2k

点赞数 1

分类专栏： Strings

本文链接：https://blog.csdn.net/crazy_ac/article/details/7850919

版权

Strings 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该博客介绍了一道关于KMP算法的题目，要求找出字符串S中能拼成目标字符串T的所有子串组合，并计算其方案数。通过计算T的前缀和后缀在S中出现的次数，可以得出答案。博主分享了解决问题的方法，即使用KMP算法更新前缀和后缀出现的次数，然后进行求和。

摘要由CSDN通过智能技术生成

我以前做过的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<cstdio>
#include<cstring>
#include<algorithm>
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());
    }
}