小蒟蒻的求助
查看原帖
小蒟蒻的求助
1241241
zbs_xj楼主2025/7/30 15:23
#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N=2e5+5,M=4e5+5,mod=998244353;
LL a[N],vis[N],sum[M];
LL ans,n;
int main(){
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> a[i];
		vis[a[i]]++;
	}
	for(int i=1;i<=M-5;i++)
		sum[i]=sum[i-1]+vis[i];
	for(int i=1;i<=N-5;i++)
		if(vis[i]>=2)
			ans=(ans+vis[i]*(vis[i]-1)*(sum[i*2-1]-vis[i])/2+vis[i]*(vis[i]-1)*(vis[i]-2)/6)%mod;
	cout << ans << endl;
    return 0;
}
2025/7/30 15:23
加载中...