結果

問題 No.1470 Mex Sum
コンテスト
ユーザー bonKotsu
提出日時 2021-04-12 22:33:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 535 bytes
コンパイル時間 1,395 ms
コンパイル使用メモリ 167,520 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-28 18:03:45
合計ジャッジ時間 4,606 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define ll long long
#define rep(i, n) for (int i = 0; i < (n); i++)
#define P pair<int, int>

int main() {
  ll n;
  cin >> n;
  vector<int> a(n);
  rep(i, n) cin >> a[i];
  ll cnt1 = 0, cnt2 = 0, others = 0;
  rep(i, n) {
    if (a[i] == 1) cnt1++;
    if (a[i] == 2) cnt2++;
  }
  others = n-cnt1-cnt2;

  ll ans = 0;
  ll num3 = cnt2*cnt1, num2 = cnt1*(cnt1-1)/2+ cnt1*others, num1 = n*(n-1)/2 - num2 - num3;
  ans += 3*num3;
  ans += 2*num2;
  ans += num1; 
  cout << ans << endl;

}
0