結果

問題 No.1268 Fruit Rush 2
ユーザー SSRS
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}
0