結果

問題 No.1268 Fruit Rush 2
ユーザー stoq
提出日時 2020-09-18 08:09:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 208 ms / 2,000 ms
コード長 931 bytes
コンパイル時間 1,752 ms
コンパイル使用メモリ 82,108 KB
最終ジャッジ日時 2025-01-14 16:04:40
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
using ll = long long;

ll calc(const vector<ll> &a)
{
  int n = a.size();
  int maxA = a.back();
  vector<ll> dp(maxA + 2, 0);
  vector<bool> exist(maxA + 2, 0);
  for (int i = 0; i < n; i++)
    exist[a[i]] = true;
  for (int i = 0; i < maxA + 2; i++)
  {
    dp[i] += exist[i];
    if (i >= 2 and exist[i - 1])
      dp[i] += dp[i - 2];
  }
  ll sum = 0;
  for (int i = 0; i < maxA + 2; i++)
    sum += dp[i];
  return sum;
}

int main()
{
  int n;
  cin >> n;
  vector<ll> a(n);
  for (int i = 0; i < n; i++)
  {
    cin >> a[i];
    a[i]--;
  }
  sort(begin(a), end(a));
  a.push_back(2e18);

  ll ans = 0, minus = a[0];
  vector<ll> v;
  for (int i = 0; i < n; i++)
  {
    v.push_back(a[i] - minus);
    if (a[i + 1] - a[i] >= 3)
    {
      ans += calc(v);
      v.clear();
      minus = a[i + 1];
    }
  }
  cout << ans << "\n";
  return 0;
}
0