結果

問題 No.2500 Products in a Range
ユーザー 👑 emthrmemthrm
提出日時 2023-09-02 18:21:31
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 3,019 bytes
コンパイル時間 1,660 ms
コンパイル使用メモリ 136,884 KB
実行使用メモリ 4,356 KB
最終ジャッジ日時 2023-10-13 18:04:08
合計ジャッジ時間 3,758 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,352 KB
testcase_01 AC 2 ms
4,352 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 1 ms
4,348 KB
testcase_04 AC 1 ms
4,352 KB
testcase_05 AC 2 ms
4,352 KB
testcase_06 AC 1 ms
4,348 KB
testcase_07 AC 2 ms
4,352 KB
testcase_08 AC 3 ms
4,352 KB
testcase_09 AC 2 ms
4,352 KB
testcase_10 AC 3 ms
4,352 KB
testcase_11 AC 3 ms
4,352 KB
testcase_12 AC 2 ms
4,348 KB
testcase_13 AC 1 ms
4,348 KB
testcase_14 AC 3 ms
4,348 KB
testcase_15 AC 2 ms
4,348 KB
testcase_16 AC 4 ms
4,352 KB
testcase_17 AC 3 ms
4,352 KB
testcase_18 AC 3 ms
4,352 KB
testcase_19 AC 4 ms
4,348 KB
testcase_20 AC 3 ms
4,348 KB
testcase_21 AC 2 ms
4,348 KB
testcase_22 AC 4 ms
4,348 KB
testcase_23 AC 4 ms
4,348 KB
testcase_24 AC 4 ms
4,348 KB
testcase_25 AC 4 ms
4,352 KB
testcase_26 AC 4 ms
4,348 KB
testcase_27 AC 4 ms
4,352 KB
testcase_28 AC 4 ms
4,356 KB
testcase_29 AC 2 ms
4,352 KB
testcase_30 AC 2 ms
4,348 KB
testcase_31 AC 2 ms
4,348 KB
testcase_32 AC 2 ms
4,352 KB
testcase_33 AC 2 ms
4,352 KB
testcase_34 AC 3 ms
4,348 KB
testcase_35 AC 2 ms
4,352 KB
testcase_36 AC 2 ms
4,352 KB
testcase_37 AC 1 ms
4,348 KB
testcase_38 AC 2 ms
4,352 KB
testcase_39 AC 1 ms
4,348 KB
testcase_40 AC 1 ms
4,356 KB
testcase_41 AC 1 ms
4,356 KB
testcase_42 AC 1 ms
4,352 KB
testcase_43 AC 2 ms
4,348 KB
testcase_44 AC 2 ms
4,352 KB
testcase_45 AC 2 ms
4,352 KB
testcase_46 AC 2 ms
4,352 KB
testcase_47 AC 4 ms
4,352 KB
testcase_48 AC 2 ms
4,348 KB
testcase_49 AC 2 ms
4,352 KB
testcase_50 AC 3 ms
4,352 KB
testcase_51 AC 3 ms
4,348 KB
testcase_52 AC 4 ms
4,348 KB
testcase_53 AC 2 ms
4,352 KB
testcase_54 AC 2 ms
4,352 KB
testcase_55 AC 3 ms
4,352 KB
testcase_56 AC 3 ms
4,348 KB
testcase_57 AC 4 ms
4,348 KB
testcase_58 AC 3 ms
4,352 KB
testcase_59 AC 5 ms
4,356 KB
testcase_60 AC 2 ms
4,352 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <bit>
#include <cassert>
#include <cstdint>
#include <functional>
#include <iostream>
#include <numeric>
#include <vector>

namespace emthrm {

template <typename Band>
struct SparseTable {
  using BinOp = std::function<Band(Band, Band)>;

  SparseTable() = default;

  explicit SparseTable(const std::vector<Band>& a, const BinOp bin_op) {
    init(a, bin_op);
  }

  void init(const std::vector<Band>& a, const BinOp bin_op_) {
    bin_op = bin_op_;
    const int n = a.size();
    assert(n > 0);
    lg.assign(n + 1, 0);
    for (int i = 2; i <= n; ++i) {
      lg[i] = lg[i >> 1] + 1;
    }
    const int table_h = std::countr_zero(std::bit_floor(a.size())) + 1;
    data.assign(table_h, std::vector<Band>(n));
    std::copy(a.begin(), a.end(), data.front().begin());
    for (int i = 1; i < table_h; ++i) {
      for (int j = 0; j + (1 << i) <= n; ++j) {
        data[i][j] = bin_op(data[i - 1][j], data[i - 1][j + (1 << (i - 1))]);
      }
    }
  }

  Band query(const int left, const int right) const {
    assert(left < right);
    const int h = lg[right - left];
    return bin_op(data[h][left], data[h][right - (1 << h)]);
  }

 private:
  BinOp bin_op;
  std::vector<int> lg;
  std::vector<std::vector<Band>> data;
};

}  // namespace emthrm

// <AC>
// 区間最大値を管理する sparse table を用いる。
int main() {
  constexpr int kMaxN = 5000, kMinA = -1000000000, kMaxA = 1000000000;

  int n, l, r;
  std::cin >> n >> l >> r;
  assert(1 <= n && n <= kMaxN && kMinA <= l && l <= r && r <= kMaxA);
  std::vector<std::int64_t> a(n);
  for (int i = 0; i < n; ++i) {
    std::cin >> a[i];
    assert(kMinA <= a[i] && a[i] <= kMaxA);
  }
  std::ranges::sort(a);

  std::vector<int> p_prime(n, n), q(n, -1), q_prime(n);
  for (int i = 0; i < n; ++i) {
    if (a[i] < 0) {
      for (int lb = -1; lb + 1 < p_prime[i];) {
        const int p = std::midpoint(lb, p_prime[i]);
        (a[i] * a[p] <= r ? p_prime[i] : lb) = p;
      }
      for (int ub = n; q[i] + 1 < ub;) {
        const int cand_q = std::midpoint(q[i], ub);
        (a[i] * a[cand_q] >= l ? q[i] : ub) = cand_q;
      }
    } else {
      for (int lb = -1; lb + 1 < p_prime[i];) {
        const int p = std::midpoint(lb, p_prime[i]);
        (a[i] * a[p] >= l ? p_prime[i] : lb) = p;
      }
      for (int ub = n; q[i] + 1 < ub;) {
        const int cand_q = std::midpoint(q[i], ub);
        (a[i] * a[cand_q] <= r ? q[i] : ub) = cand_q;
      }
    }
    if (q[i] < p_prime[i]) {
      p_prime[i] = n;
      q[i] = -1;
    } else {
      p_prime[i] = std::max(p_prime[i], i + 1);
    }
    q_prime[i] = std::min(q[i], i - 1);
  }

  int ans = 1;
  const emthrm::SparseTable<int> sparse_table(
      q_prime, [](const int x, const int y) -> int { return std::max(x, y); });
  for (int i = 0; i < n; ++i) {
    if (p_prime[i] <= q[i]) {
      const int f = 3 - p_prime[i] + sparse_table.query(p_prime[i], q[i] + 1);
      ans = std::max({ans, f, 2});
    }
  }
  std::cout << ans << '\n';
  return 0;
}
0