結果

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

ソースコード

diff # 

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N;
  cin >> N;
  llvec a(N);
  rep(i, N){
    cin >> a[i];
  }
  sort(ALL(a));
  ll ans = 0;
  rep(i, N){
    if(a[i]==1){
      ll ind = upper_bound(ALL(a), 2) - lower_bound(ALL(a), 2);//2の数
      ll ind2 = a.end() - lower_bound(ALL(a), 3);//3以上の数
      ll ind3 = upper_bound(ALL(a), 1) - a.begin();//1の数
      ans += 3*ind + ind2*2 + (ind3-i-1)*2;
    }else{
      ans += 1*(N-i-1);
    }
  }
  cout << ans << endl;
}
0