結果
問題 | No.1470 Mex Sum |
ユーザー |
|
提出日時 | 2021-04-09 21:51:44 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 51 ms / 2,000 ms |
コード長 | 1,197 bytes |
コンパイル時間 | 2,091 ms |
コンパイル使用メモリ | 170,060 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-06-25 05:07:50 |
合計ジャッジ時間 | 5,220 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
#include "bits/stdc++.h" using namespace std; using ll = long long; void Main() { ll N; cin >> N; vector<ll> A(N, 0); for (ll i = 0; i < N; ++i) { cin >> A[i]; } vector<ll> cnt1(N + 1, 0), cnt2(N + 1, 0); cnt1[0] = 0; cnt2[0] = 0; for (ll i = 1; i < N + 1; ++i) { cnt1[i] = cnt1[i - 1] + (A[i - 1] == 1 ? 1 : 0); cnt2[i] = cnt2[i - 1] + (A[i - 1] == 2 ? 1 : 0); } vector<ll> num1From(N, 0), num2From(N, 0); for (ll i = 0; i < N; ++i) { num1From[i] = cnt1.back() - cnt1[i]; num2From[i] = cnt2.back() - cnt2[i]; } ll ans = 0LL; for (ll i = 0; i < N - 1; ++i) { if (A[i] == 1) { ll n = num2From[i + 1]; ans += n * 3; ans += (N - i - 1 - n) * 2; } else if (A[i] == 2) { ll n = num1From[i + 1]; ans += n * 3; ans += (N - i - 1 - n) * 1; } else { ll n = num1From[i + 1]; ans += n * 2; ans += (N - i - 1 - n) * 1; } } cout << ans << endl; } int main() { std::cout << std::fixed << std::setprecision(15); Main(); return 0; }