Skip to content

sparse_table.hpp

SECTIONData Structure INCLUDEnoya/ds/sparse_table.hpp

数组不再修改且区间运算幂等(如最小值、最大值或 gcd)时,用 \(O(n\log n)\) 预处理把每次区间查询降到 \(O(1)\)。

Complexity: Time: O(n log n) build and O(1) idempotent range query. Space: O(n log n).

AC 记录:staticrmq。

跳到代码 · GitHub ↗

Implementation

当前头文件,省略 include guard;依赖见 #include。

/// @complexity Time: O(n log n) build and O(1) idempotent range query.
/// Space: O(n log n).

#include <bit>
#include <cassert>
#include <utility>
#include <vector>

namespace noya {

/// @brief Static O(1) range product for an idempotent semilattice.
/// Monoid provides value_type, unit(), and an associative, idempotent op().
/// Empty ranges return unit(); input arrays are owned by the table.
/// @details For k = floor(log2(r-l)), [l, r) is covered by the length-2^k
/// blocks beginning at l and ending at r. They may overlap, so combining them
/// is correct precisely because op(x, x) = x.
template <class Monoid> class sparse_table {

public:
  using value_type = typename Monoid::value_type;

  sparse_table() = default;

  explicit sparse_table(std::vector<value_type> a) { build(std::move(a)); }

  template <class F> sparse_table(int n_, F f) { build(n_, std::move(f)); }

  template <class F> void build(int n_, F f) {
    assert(n_ >= 0);
    std::vector<value_type> a;
    a.reserve(n_);
    for (int i = 0; i < n_; i++) {
      a.push_back(f(i));
    }
    build(std::move(a));
  }

  /// @brief Rebuild from a static array.
  void build(std::vector<value_type> a) {
    n = int(a.size());
    int lg = int(std::bit_width(static_cast<unsigned>(n)));
    st.resize(lg);
    if (n == 0) {
      return;
    }

    st[0] = std::move(a);
    for (int dep = 1; dep < lg; dep++) {
      int w = 1 << dep;
      int hf = w >> 1;
      st[dep].resize(n - w + 1, Monoid::unit());
      for (int l = 0; l + w <= n; l++) {
        st[dep][l] = Monoid::op(st[dep - 1][l], st[dep - 1][l + hf]);
      }
    }
  }

  int size() const { return n; }
  bool empty() const { return n == 0; }

  value_type get(int p) const {
    assert(0 <= p && p < n);
    return st[0][p];
  }

  value_type all_prod() const { return prod(0, n); }

  /// @brief Return the idempotent product over [l, r).
  value_type prod(int l, int r) const {
    assert(0 <= l && l <= r && r <= n);
    if (l == r) {
      return Monoid::unit();
    }
    int dep = int(std::bit_width(static_cast<unsigned>(r - l))) - 1;
    int w = 1 << dep;
    return Monoid::op(st[dep][l], st[dep][r - w]);
  }

private:
  int n = 0;
  std::vector<std::vector<value_type>> st;
};

} // namespace noya

 // NOYA_SPARSE_TABLE_HPP
#ifndef NOYA_SPARSE_TABLE_HPP
#define NOYA_SPARSE_TABLE_HPP 1

/// @complexity Time: O(n log n) build and O(1) idempotent range query.
/// Space: O(n log n).

#include <bit>
#include <cassert>
#include <utility>
#include <vector>

namespace noya {

/// @brief Static O(1) range product for an idempotent semilattice.
/// Monoid provides value_type, unit(), and an associative, idempotent op().
/// Empty ranges return unit(); input arrays are owned by the table.
/// @details For k = floor(log2(r-l)), [l, r) is covered by the length-2^k
/// blocks beginning at l and ending at r. They may overlap, so combining them
/// is correct precisely because op(x, x) = x.
template <class Monoid> class sparse_table {

public:
  using value_type = typename Monoid::value_type;

  sparse_table() = default;

  explicit sparse_table(std::vector<value_type> a) { build(std::move(a)); }

  template <class F> sparse_table(int n_, F f) { build(n_, std::move(f)); }

  template <class F> void build(int n_, F f) {
    assert(n_ >= 0);
    std::vector<value_type> a;
    a.reserve(n_);
    for (int i = 0; i < n_; i++) {
      a.push_back(f(i));
    }
    build(std::move(a));
  }

  /// @brief Rebuild from a static array.
  void build(std::vector<value_type> a) {
    n = int(a.size());
    int lg = int(std::bit_width(static_cast<unsigned>(n)));
    st.resize(lg);
    if (n == 0) {
      return;
    }

    st[0] = std::move(a);
    for (int dep = 1; dep < lg; dep++) {
      int w = 1 << dep;
      int hf = w >> 1;
      st[dep].resize(n - w + 1, Monoid::unit());
      for (int l = 0; l + w <= n; l++) {
        st[dep][l] = Monoid::op(st[dep - 1][l], st[dep - 1][l + hf]);
      }
    }
  }

  int size() const { return n; }
  bool empty() const { return n == 0; }

  value_type get(int p) const {
    assert(0 <= p && p < n);
    return st[0][p];
  }

  value_type all_prod() const { return prod(0, n); }

  /// @brief Return the idempotent product over [l, r).
  value_type prod(int l, int r) const {
    assert(0 <= l && l <= r && r <= n);
    if (l == r) {
      return Monoid::unit();
    }
    int dep = int(std::bit_width(static_cast<unsigned>(r - l))) - 1;
    int w = 1 << dep;
    return Monoid::op(st[dep][l], st[dep][r - w]);
  }

private:
  int n = 0;
  std::vector<std::vector<value_type>> st;
};

} // namespace noya

#endif // NOYA_SPARSE_TABLE_HPP
/// @complexity Time: O(n log n) build and O(1) idempotent range query.
/// Space: O(n log n).

#include <bit>
#include <cassert>
#include <utility>
#include <vector>

namespace noya {

/// @brief Static O(1) range product for an idempotent semilattice.
/// Monoid provides value_type, unit(), and an associative, idempotent op().
/// Empty ranges return unit(); input arrays are owned by the table.
/// @details For k = floor(log2(r-l)), [l, r) is covered by the length-2^k
/// blocks beginning at l and ending at r. They may overlap, so combining them
/// is correct precisely because op(x, x) = x.
template <class Monoid> class sparse_table {

public:
  using value_type = typename Monoid::value_type;

  sparse_table() = default;

  explicit sparse_table(std::vector<value_type> a) { build(std::move(a)); }

  template <class F> sparse_table(int n_, F f) { build(n_, std::move(f)); }

  template <class F> void build(int n_, F f) {
    assert(n_ >= 0);
    std::vector<value_type> a;
    a.reserve(n_);
    for (int i = 0; i < n_; i++) {
      a.push_back(f(i));
    }
    build(std::move(a));
  }

  /// @brief Rebuild from a static array.
  void build(std::vector<value_type> a) {
    n = int(a.size());
    int lg = int(std::bit_width(static_cast<unsigned>(n)));
    st.resize(lg);
    if (n == 0) {
      return;
    }

    st[0] = std::move(a);
    for (int dep = 1; dep < lg; dep++) {
      int w = 1 << dep;
      int hf = w >> 1;
      st[dep].resize(n - w + 1, Monoid::unit());
      for (int l = 0; l + w <= n; l++) {
        st[dep][l] = Monoid::op(st[dep - 1][l], st[dep - 1][l + hf]);
      }
    }
  }

  int size() const { return n; }
  bool empty() const { return n == 0; }

  value_type get(int p) const {
    assert(0 <= p && p < n);
    return st[0][p];
  }

  value_type all_prod() const { return prod(0, n); }

  /// @brief Return the idempotent product over [l, r).
  value_type prod(int l, int r) const {
    assert(0 <= l && l <= r && r <= n);
    if (l == r) {
      return Monoid::unit();
    }
    int dep = int(std::bit_width(static_cast<unsigned>(r - l))) - 1;
    int w = 1 << dep;
    return Monoid::op(st[dep][l], st[dep][r - w]);
  }

private:
  int n = 0;
  std::vector<std::vector<value_type>> st;
};

} // namespace noya

 // NOYA_SPARSE_TABLE_HPP