全wa求助
查看原帖
全wa求助
1271310
mysb楼主2025/1/20 14:59
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=2e5+10,mod=998244353;
int n,m;
int a[N];
int v[N],w[N],sum,mx,mmx;
int t=1,p,q;
int l,r,ans=1;
int mp[N],s[N];
signed main(){
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i],mp[a[i]]++;
	for(int i=1;i<=200050;i++){
		s[i]=(s[i-1]+mp[i])%mod;
	}
	for(int i=1;i<=200050;i++){
		v[i]=mp[i]*(mp[i]-1)*(mp[i]-2)/6;
		v[i]%=mod;
	}
	for(int i=1;i<=200050;i++){
		if(mp[i]>=2){
			int w=(mp[i]*(mp[i]-1)/2);
			int k=w*(s[2*i-1]-mp[i]);
			p=(p+k)%mod;
			p=(p+v[mp[i]])%mod;
			p%=mod;
		}
	}
	p%=mod;
	cout<<p;
	return 0;
}
2025/1/20 14:59
加载中...