結果

問題 No.1470 Mex Sum
ユーザー soto800
提出日時 2021-04-09 22:16:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,298 bytes
コンパイル時間 1,589 ms
コンパイル使用メモリ 165,784 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-25 05:48:45
合計ジャッジ時間 3,518 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

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;
	lli d[3]={0};
	REP(i,0,n){
		lli b;
		cin>>b;
		if(b==1 || b==2)d[b]++;
		else d[0]++;
	}
	lli ans = 0;

	ans = 3*d[1]*d[2] + 2 * (d[1]*d[0] + d[1]*(d[1]-1)/2) + d[2]*d[0] + d[2]*(d[2]-1)/2 + d[0]*(d[0]-1)/2;


	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