結果
問題 | No.1268 Fruit Rush 2 |
ユーザー |
![]() |
提出日時 | 2020-10-23 22:33:00 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 254 ms / 2,000 ms |
コード長 | 792 bytes |
コンパイル時間 | 2,335 ms |
コンパイル使用メモリ | 183,580 KB |
実行使用メモリ | 25,248 KB |
最終ジャッジ日時 | 2024-07-21 11:21:29 |
合計ジャッジ時間 | 7,136 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main(){ int N; cin >> N; vector<long long> A(N); for (int i = 0; i < N; i++){ cin >> A[i]; } sort(A.rbegin(), A.rend()); set<long long> st; for (int i = 0; i < N; i++){ st.insert(A[i]); } vector<vector<vector<long long>>> P(2); for (int i = 0; i < N; i++){ int r = A[i] % 2; if (st.count(A[i] + 2)){ P[r].back().push_back(A[i]); } else { P[r].push_back(vector<long long>(1, A[i])); } } long long ans = N; for (int i = 0; i < 2; i++){ int cnt = P[i].size(); for (int j = 0; j < cnt; j++){ int sz = P[i][j].size(); for (int k = 0; k < sz; k++){ if (st.count(P[i][j][k] - 1)){ ans += k + 1; } } } } cout << ans << endl; }