結果
問題 | No.1470 Mex Sum |
ユーザー |
![]() |
提出日時 | 2021-04-09 22:07:04 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 17 ms / 2,000 ms |
コード長 | 1,345 bytes |
コンパイル時間 | 3,525 ms |
コンパイル使用メモリ | 229,964 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-06-25 05:37:00 |
合計ジャッジ時間 | 5,926 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
#include <bits/stdc++.h> using namespace std; using lint = long long int; using P = pair<int, int>; using PL = pair<lint, lint>; #define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++) #define IFOR(i, begin, end) for(int i=(end)-1,i##_begin_=(begin);i>=i##_begin_;i--) #define REP(i, n) FOR(i,0,n) #define IREP(i, n) IFOR(i,0,n) #define ALL(a) (a).begin(),(a).end() constexpr int MOD = 1000000007; constexpr lint B1 = 1532834020; constexpr lint M1 = 2147482409; constexpr lint B2 = 1388622299; constexpr lint M2 = 2147478017; constexpr int INF = 2147483647; void yes(bool expr) {cout << (expr ? "YES" : "NO") << "\n";} template<class T>void chmax(T &a, const T &b) { if (a<b) a=b; } template<class T>void chmin(T &a, const T &b) { if (b<a) a=b; } #include <atcoder/all> using namespace atcoder; int main() { ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); lint N; cin >> N; vector<lint> A(N); REP(i, N) cin >> A[i]; lint ans = 0; lint ones = 0; lint twos = 0; lint other = 0; REP(i, N) { if(A[i] == 1) { ans += 3 * twos; ans += 2 * (ones + other); ones++; } else if(A[i] == 2) { ans += 3 * ones; ans += 1 * (twos + other); twos++; } else { ans += 2 * ones; ans += 1 * (twos + other); other++; } } cout << ans << endl; }