結果
問題 |
No.1470 Mex Sum
|
ユーザー |
|
提出日時 | 2021-04-09 21:28:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 46 ms / 2,000 ms |
コード長 | 1,312 bytes |
コンパイル時間 | 1,928 ms |
コンパイル使用メモリ | 195,780 KB |
最終ジャッジ日時 | 2025-01-20 13:34:20 |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define rep(i,n) for(ll i=0;i<n;i++) #define repl(i,l,r) for(ll i=(l);i<(r);i++) #define per(i,n) for(ll i=(n)-1;i>=0;i--) #define perl(i,r,l) for(ll i=r-1;i>=l;i--) #define fi first #define se second #define pb push_back #define ins insert #define pqueue(x) priority_queue<x,vector<x>,greater<x>> #define all(x) (x).begin(),(x).end() #define CST(x) cout<<fixed<<setprecision(x) #define rev(x) reverse(x); using ll=long long; using vl=vector<ll>; using vvl=vector<vector<ll>>; using pl=pair<ll,ll>; using vpl=vector<pl>; using vvpl=vector<vpl>; const ll MOD=1000000007; const ll MOD9=998244353; const int inf=1e9+10; const ll INF=4e18; const ll dy[8]={1,0,-1,0,1,1,-1,-1}; const ll dx[8]={0,1,0,-1,1,-1,1,-1}; template <typename T> inline bool chmax(T &a, T b) { return ((a < b) ? (a = b, true) : (false)); } template <typename T> inline bool chmin(T &a, T b) { return ((a > b) ? (a = b, true) : (false)); } int main(){ ll n;cin >> n; vl v(n);rep(i,n)cin >> v[i]; vl cnt(3); rep(i,n){ if(v[i]==1)cnt[0]++; else if(v[i]==2)cnt[1]++; else cnt[2]++; } ll ans=0; ans+=cnt[0]*(cnt[0]-1)/2*2; ans+=cnt[0]*cnt[1]*3; ans+=cnt[0]*cnt[2]*2; ans+=cnt[1]*(cnt[1]-1)/2; ans+=cnt[1]*cnt[2]; ans+=cnt[2]*(cnt[2]-1)/2; cout << ans <<endl; }