acwing160 匹配统计

做法1:哈希+二分 时间复杂度O(nlogm+q)

做法2:kmp匹配 时间复杂度O(n+m+q)

值得思考的做法2
cnt[j]表示匹配长度为j的个数
一位一位的kmp匹配 若我们发现到a的第i位有b的前j个与它匹配 就++cnt[j]

kmp后倒着遍历 cnt[nxt[j]]+=cnt[j],因为kmp每一位匹配到的都是当前最长的,这一遍可以把短的也加进来

但这样会有一个问题 若一个a的后缀与b最长匹配长度为k则在k-1长度被多统计了 所以经过以上的计算后cnt[j]表示的其实是大于等于j长度的个数 输出答案时要改成cnt[j]-cnt[j+1]

代码

#include<iostream>
#include<cstdio>

using namespace std;
int la,lb;
char a[200005],b[200005];
int nxt[200005];
int cnt[200005];
int main()
{
	int q;
	scanf("%d%d%d",&la,&lb,&q);
	scanf("%s",a+1);
	scanf("%s",b+1);
	for(int i=2,j=0;i<=lb;++i)
	{
		while(j&&b[j+1]!=b[i])j=nxt[j];
		if(b[j+1]==b[i])++j;
		nxt[i]=j;
	}
	int j=0;
	for(int i=1;i<=la;++i)
	{
		while(j&&b[j+1]!=a[i])j=nxt[j];
		if(b[j+1]==a[i])++j;
		++cnt[j];
		if(j==lb)
		{
			j=nxt[j];
		}
	}
	for(int i=lb;i>=1;--i)cnt[nxt[i]]+=cnt[i];
	int x;
	while(q--)
	{
		scanf("%d",&x);
		printf("%d\n",cnt[x]-cnt[x+1]);
	}
	return 0; 
}
发布了25 篇原创文章 · 获赞 1 · 访问量 314

猜你喜欢

转载自blog.csdn.net/qq_43665203/article/details/104243849