結果

問題 No.1470 Mex Sum
コンテスト
ユーザー soto800
提出日時 2021-04-09 22:09:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,258 bytes
コンパイル時間 1,626 ms
コンパイル使用メモリ 176,772 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-25 05:42:14
合計ジャッジ時間 4,130 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 14 WA * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
 
#define lli long long int
#define REP(i,s,n) for(lli i=s;i<n;i++)
#define INF (1LL<<62)
#define DEBUG 0
#define mp(a,b) make_pair(a,b)
#define SORT(V) sort(V.begin(),V.end())
#define PI (3.141592653589794)
#define TO_STRING(VariableName) # VariableName
#define LOG(x) if(DEBUG)cout<<TO_STRING(x)<<"="<<x<<" "<<endl;
#define LOG2(x,y) if(DEBUG)cout<<TO_STRING(x)<<"="<<x<<" "<<TO_STRING(y)<<"="<<y<<endl;
#define LOG3(x,y,z) if(DEBUG)cout<<TO_STRING(x)<<"="<<x<<" "<<TO_STRING(y)<<"="<<y<<" "<<TO_STRING(z)<<"="<<z<<endl;
#define LOG4(w,x,y,z) if(DEBUG)cout<<TO_STRING(w)<<"="<<w<<" "<<TO_STRING(x)<<"="<<x<<" "<<TO_STRING(y)<<"="<<y<<" "<<TO_STRING(z)<<"="<<z<<endl;
 
template<class T>bool chmax(T & a, const T & b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T& a, const T& b) { if (b < a) { a = b; return 1; } return 0; }

void solve(){

	lli n;
	cin>>n;
	map<lli,lli> a;
	REP(i,0,n){
		lli b;
		cin>>b;
		a[b]++;
	}
	lli ans = 0;
	lli d = n*(n-1)/2 - (a[1]*a[2] + a[1]*(a[1]-1)/2);
	ans = 3*a[1]*a[2] + 2*a[1]*(a[1]-1)/2 + d;
	cout<<ans<<endl;

}
 
int main(){
 
	ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
 
	lli t = 1;
	//cin>>t;
	REP(i,0,t)solve();
 
	return 0;
}
0