結果

問題 No.1470 Mex Sum
コンテスト
ユーザー beet
提出日時 2021-04-09 21:33:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 18 ms / 2,000 ms
コード長 839 bytes
コンパイル時間 1,899 ms
コンパイル使用メモリ 194,020 KB
最終ジャッジ日時 2025-01-20 13:40:56
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using Int = long long;
const char newl = '\n';

template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
template<typename T> void drop(const T &x){cout<<x<<endl;exit(0);}
template<typename T=Int>
vector<T> read(size_t n){
  vector<T> ts(n);
  for(size_t i=0;i<n;i++) cin>>ts[i];
  return ts;
}

//INSERT ABOVE HERE
signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  Int n;
  cin>>n;
  auto as=read(n);

  Int one=count(as.begin(),as.end(),1);
  Int two=count(as.begin(),as.end(),2);

  Int ans=0;
  // three: one, two
  ans+=3*one*two;
  // one: not one
  ans+=1*(n-one)*(n-one-1)/2;
  // two: others
  ans+=2*(n*(n-1)/2-one*two-(n-one)*(n-one-1)/2);
  cout<<ans<<newl;
  return 0;
}
0