結果
問題 | No.1470 Mex Sum |
ユーザー |
|
提出日時 | 2021-04-09 22:04:38 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 130 ms / 2,000 ms |
コード長 | 1,206 bytes |
コンパイル時間 | 7,060 ms |
コンパイル使用メモリ | 252,332 KB |
最終ジャッジ日時 | 2025-01-20 14:25:12 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>typedef unsigned long long ULLONG;typedef long long LLONG;static const LLONG MOD_NUM = 1000000007LL; //998244353LL;static const LLONG INF_NUM = LLONG_MAX / 20LL;template<class _T> static void get(_T& a) {std::cin >> a;}template<class _T> static void get(_T& a, _T& b) {std::cin >> a >> b;}template<class _T> static void get(_T& a, _T& b, _T& c) {std::cin >> a >> b >> c;}template <class _T> static _T tp_abs(_T a) {if (a < (_T)0) {a *= (_T)-1;}return a;}static void task();int main(){task();fflush(stdout);return 0;}static void task(){int N;get(N);std::vector<LLONG> ai(N);for (int i = 0; i < N; i++) {get(ai[i]);}std::sort(ai.begin(), ai.end());int oneOver = N, twoOver = N;for (int i = 0; i < N; i++) {if (ai[i] > 1) {if(oneOver == N) oneOver = i;}if (ai[i] > 2) {if (twoOver == N) twoOver = i;}}LLONG ans = 0;for (int i = 0; i < N - 1; i++) {if (i < oneOver) {ans += 2LL * (LLONG)(oneOver - i - 1);ans += 3LL * (LLONG)(twoOver - oneOver);ans += 2LL * (LLONG)(N - twoOver);}else {ans += (LLONG)(N - i - 1);}}printf("%lld\n", ans);}