hdu 6153 (扩展kmp)

题意:给定两个字符s1,(length=len1),s2(length=len2);求s2[i->len](i=1,2,3,4,....len),在s1中出现的次数乘上其长度;例如 s1=abababc,s2=abc,则应求abc,bc,c,在s1中出现的次

数。然而扩展kmp是求前缀的,所以想到将s1,s2反转一下,即求cba,cb,c,在cbababa中的出现次数答案与不反转相同(仔细想想就知道),要想求cba,cb,c,在cbababa中的出现

次数,只需求以cbababa为母串,以cba为字串的ex数组因为 cba包括(cb,c),最后每一个不为0的ex[i]利用等差数列求和公式算一遍加到ans中(cba出现一次,也就是3*1,

那么相当于cb出现了一次,即2*1,也相当于c出现一次即1*1,最后总次数加上1+2+3)。

  下面是ac代码,代码中的扩展kmp模板百度到处是,扩展kmp不懂的话推荐百度的刘雅琼老师的ppt。https://wenku.baidu.com/view/8e9ebefb0242a8956bece4b3.html

#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const long long maxn=1e6+5;
const long long  mod=1e9+7;
ll ex[maxn],nex[maxn];
char s1[maxn],s2[maxn];
void get_next(char *str)
{
	int len,j,i=0,po;
	len=nex[0]=strlen(str);
	while (str[i]==str[i+1]&&i+1<len)
	i++;
	nex[1]=i;
	po=1;
	for (i=2;i<len;i++)
	{
		if (nex[i-po]+i<nex[po]+po)
		nex[i]=nex[i-po];
		else
		{
			j=nex[po]+po-i;
			if (j<0) j=0;
			while (i+j<len&&str[j]==str[j+i])
			j++;
			nex[i]=j;
			po=i;
		}
	}
}
void get_extend(char *s1,char *s2)
{
	int i=0,j,po,len,l2;
	len=strlen(s1);
	l2=strlen(s2);
	get_next(s2);
	while (s1[i]==s2[i]&&i<len&&i<l2)
	i++;
	ex[0]=i;
	po=0;
	for (i=1;i<len;i++)
	{
	   if (nex[i-po]+i<ex[po]+po)
	   ex[i]=nex[i-po];
	   else
	   {
	   	   j=ex[po]+po-i;
	   	   if (j<0) j=0;
	   	   while (i+j<len&&s2[j]==s1[i+j]&&j<l2)
	   	   j++;
	   	   ex[i]=j;
	   	   po=i;
	   }
	}
}
int main()
{
	int n1,n2,t;
	ll ans;
	cin>>t;
	while (t--)
	{
	  scanf ("%s%s",s1,s2);
	  ans=0;
	  n1=strlen(s1);
	  n2=strlen(s2);
	  reverse(s1,s1+n1);
	  reverse(s2,s2+n2);
	  memset(nex,0,sizeof(nex));
	  memset(ex,0,sizeof(ex));
	  get_extend(s1,s2);
	  for (int i=0;i<n1;i++)
	  if (ex[i])
	  ans=(ans+ex[i]*(ex[i]+1)/2)%mod;
	  cout<<ans<<endl;
      }
} 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值