n方做法,sub2仍然WA了几个点QWQ
查看原帖
n方做法,sub2仍然WA了几个点QWQ
160904
happy_lfz楼主2020/7/27 16:02
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 3000005
#define mod 4294967296ll
#define int long long
using namespace std;

int la,lb,mx,tmp,res,p[N],nxt[N],s[N];
char a[N],b[N];

//int fi(int l,int r)
//{
//	int j=0,ans=0;
//	for(int i=l;i<=r;i++)
//	{
//		while(j&&a[i]!=b[j+1])j=nxt[j];
//		if(a[i]==b[j+1])j++;
//		if(j==lb)ans++,j=nxt[j];
//	}
//	return ans;
//}
signed main()
{
	scanf("%lld%lld",&la,&lb);
	scanf("%s",a+1);scanf("%s",b+1);
	
	int j=0;
	for(int i=2;i<=lb;i++)
	{
		while(j&&b[i]!=b[j+1])j=nxt[j];
		if(b[i]==b[j+1])j++;
		nxt[i]=j;
	}
	j=0;
	for(int i=1;i<=la;i++)
	{
		while(j&&a[i]!=b[j+1])j=nxt[j];
		if(a[i]==b[j+1])j++;
		if(j==lb)s[i]=1,j=nxt[j];
	}
	
	for(int i=1;i<=la;i++)s[i]=(s[i]+s[i-1])%mod;
//	
//	for(int i=1;i<=la;i++)printf("\n%d:%d",i,s[i]);
	
	for(int i=1;i<=la;i++)
	{
		p[i]=mx>i?min(p[2*tmp-i],mx-i):1;
		while(a[i+p[i]]==a[i-p[i]])p[i]++;
		if(i+p[i]>mx)mx=i+p[i],tmp=i;
	}
	
	for(int i=1;i<=la;i++)
		while(2*p[i]-1>=lb)
			res=(res+s[i+p[i]-1]-s[i+lb-p[i]-1]+mod)%mod,p[i]--; 
//			res+=fi(i-p[i]+1,i+p[i]-1),p[i]--;
	printf("%lld\n",res);
	return 0;
}
2020/7/27 16:02
加载中...