結果
問題 | No.1470 Mex Sum |
ユーザー |
|
提出日時 | 2021-04-09 22:51:02 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 64 ms / 2,000 ms |
コード長 | 879 bytes |
コンパイル時間 | 2,857 ms |
コンパイル使用メモリ | 191,208 KB |
最終ジャッジ日時 | 2025-01-20 15:02:02 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <math.h> #include <iomanip> #include <cstdint> #include <string> #include <sstream> template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } #define rep(i,n) for (int i = 0; i < (n); ++i) typedef long long ll; using P=pair<int,int>; const int INF=1001001001; const int mod=1e9+7; int main(){ int n; cin>>n; ll cnt1=0,cnt2=0; rep(i,n){ int a; cin>>a; if(a==1){cnt1++;} if(a==2){cnt2++;} } ll ans=0; ll p=n-cnt1,p2=n-cnt1-cnt2; ans+=p*(p-1)/2; ans+=2*cnt1*(cnt1-1)/2+2*cnt1*p2; ans+=3*cnt1*cnt2; cout<<ans<<endl; return 0; }