QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#397646#5440. P-P-PalindromeDiuWA 17ms58660kbC++141.8kb2024-04-24 15:33:472024-04-24 15:33:48

Judging History

你现在查看的是最新测评结果

  • [2024-04-24 15:33:48]
  • 评测
  • 测评结果:WA
  • 用时:17ms
  • 内存:58660kb
  • [2024-04-24 15:33:47]
  • 提交

answer

// 
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e6+10,bas=131,mod=998244353;
int n;
char s[N];
int hs1[N],hs2[N],t[N],inv[N];
map<int,int> mp1[N],mp2;
int qpow(int x,int y=mod-2){
	int m=1;
	for(;y;y>>=1,x=1ll*x*x%mod)if(y&1)m=1ll*m*x%mod;
	return m;
}
int Hs1(int l,int r){
	return (hs1[r]-1ll*hs1[l-1]*t[r-l+1]%mod+mod)%mod;
}
int Hs2(int l,int r){
	return 1ll*(hs2[r]-hs2[l-1]+mod)%mod*inv[l-1]%mod;
}
bool chk(int l,int r){
	return Hs1(l,r)==Hs2(l,r);
}
signed main(){
	scanf("%d",&n);
	t[0]=1;
	for(int i=1;i<N;i++)t[i]=1ll*t[i-1]*bas%mod;
	inv[N-1]=qpow(t[N-1]);
	for(int i=N-2;i>=0;i--)inv[i]=1ll*inv[i+1]*bas%mod;
	ll ans=0;
	for(;n--;){
		scanf("\n%s",s+1);
		int len=strlen(s+1);
		for(int i=1;i<=len;i++)hs1[i]=(1ll*bas*hs1[i-1]+s[i])%mod;
		for(int i=1;i<=len;i++)hs2[i]=(hs2[i-1]+1ll*t[i-1]*s[i])%mod;
		for(int i=1;i<=len;i++){
			int l=0,r=min(i-1,len-i)+1;
			while(l+1<r){
				int mid=l+r>>1;
				if(chk(i-mid,i+mid))l=mid;
				else r=mid;
			}
//			printf("Odd:%d %d\n",i,l);
			for(;l>=0;l--){
				int h=Hs1(i-l,i+l);
				if(mp1[2*l+1][h])continue;
//				printf("%d %d\n",i-l,i+l);
				mp1[2*l+1][h]=1;
				int val=1ll*h*qpow(mod+1-t[2*l+1])%mod;
				int res=++mp2[val];
				ans+=1ll*res*res,ans-=1ll*(res-1)*(res-1);
			}
		}
		for(int i=1;i<len;i++){
			int l=0,r=min(i-1,len-i-1)+1;
			while(l+1<r){
				int mid=l+r>>1;
				if(chk(i-mid,i+1+mid))l=mid;
				else r=mid;
			}
//			printf("Even:%d %d\n",i,l);
			for(;l>=0;l--){
				int h=Hs1(i-l,i+1+l);
				if(mp1[2*l+2][h])continue;
//				printf("%d %d\n",i-l,i+1+l);
				mp1[2*l+2][h]=1;
				int val=1ll*h*qpow(mod+1-t[2*l+2])%mod;
				int res=++mp2[val];
				ans+=1ll*res*res,ans-=1ll*(res-1)*(res-1);
			}
		}
	}
	printf("%lld\n",ans);
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 15ms
memory: 58660kb

input:

2
aaaa
aaa

output:

16

result:

ok 1 number(s): "16"

Test #2:

score: -100
Wrong Answer
time: 17ms
memory: 58596kb

input:

3
abaaa
abbbba
bbbaba

output:

30

result:

wrong answer 1st numbers differ - expected: '28', found: '30'