为啥这题开O2我就会RE?
查看原帖
为啥这题开O2我就会RE?
29093
Deep_KevinLILDOGDOG楼主2020/9/9 08:29
#include<bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;

const int N=3000010;
int fail[N],mx[N],tot[N][2];
long long sum[N][2];
char s[N],t[N];
int n,m;
vector<int> V[N];

int add(int x,int op,int t){
	for(int i=(op?n-x+1:x);i<=n;i+=lowbit(i)) 
		sum[i][op]+=x*t,tot[i][op]+=t;
}
long long gs(int x,int op,int b){
	long long ans=0,we=0;
	for(int i=(op?n-x+1:x);i;i-=lowbit(i)) 
		ans+=sum[i][op],we+=tot[i][op];
	return ans+we*b;
}

void gn(){
	fail[0]=-1;
	for(int i=1;i<=m;i++){
		int now=fail[i-1];
		while(now!=-1 && t[now+1]!=t[i]) now=fail[now];
		fail[i]=now+1;
	}
}

void kmp(){
	int i=1,j=1;
	for(int i=1,j=1;i<=n;i++,j+=2){
		j--;
		while(j!=-1 && t[j+1]!=s[i]) j=fail[j];
		if(j==m-1){ 
			add(i,0,1);
			if(m-1) V[(i+i-m+1)/2+1].push_back(i);
		}
	}
}

void mnc(){
	int mr=0,op=0;
	long long ans=0;
	for(int i=1;i<=n;i++){
		if(i<mr) mx[i]=min(mr-i,mx[2*op-i]);
		while(i-mx[i]>1 && i+mx[i]<n && s[i-mx[i]-1]==s[i+mx[i]+1]) mx[i]++;
		if(i+mx[i]>mr) mr=i+mx[i],op=i;
		for(int j=0;j<V[i].size();j++){
			add(V[i][j],0,-1);
			add(V[i][j]-m+1,1,1);
		}
		ans+=gs(i-mx[i],1,-i+mx[i]+1);
		ans-=gs(i+mx[i],0,-i-mx[i]-1);
	}
	printf("%lld\n",ans);
}

int main(){
	scanf("%d %d",&n,&m);
	scanf("%s %s",s+1,t+1);
	gn();kmp();mnc();
}
2020/9/9 08:29
加载中...