WaSubtask#4求助
查看原帖
WaSubtask#4求助
313716
EgLund楼主2020/8/20 23:13
#include<iostream>
#include<cmath>
#include<cstring>
#define MOD(x) x%998244353
#define int long long
using namespace std;
namespace T2
{
	int n,Tong[200008<<1],Tongqzh[200008<<1]={0};
	int ans=0;
	signed main()
	{
		memset(Tong,0,sizeof(Tong));
		cin>>n;for(int i=1;i<=n;i++){int x;cin>>x;Tong[x]++;Tong[x]=MOD(Tong[x]);}
		for(int i=1;i<=400000;i++)Tongqzh[i]=MOD(Tongqzh[i-1]+Tong[i]);
		for(int i=1;i<=200000;i++)if(Tong[i]>=2)ans+=MOD(Tong[i]*(Tong[i]-1)*(Tongqzh[2*i-1]-Tong[i])/2+Tong[i]*(Tong[i]-1)*(Tong[i]-2)/6);
		cout<<ans;
	}
}
signed main()
{
	T2::main();
}

code^

2020/8/20 23:13
加载中...