Linear-RMQ (structure/others/linear-rmq.hpp)

• View this file on GitHub
• Last update: 2024-04-27 01:27:28+09:00
• Include: #include "structure/others/linear-rmq.hpp"

Verified with

• test/verify/yosupo-staticrmq-5.test.cpp

Code

/**
 * @brief Linear-RMQ
 **/
template <typename Comp>
struct LinearRMQ {
  vector<int> small;
  vector<vector<int> > large;

  LinearRMQ() = default;

  Comp comp;

  static inline int msb(int c) { return 31 - __builtin_clz(c); }

  static inline int ctz(int c) { return __builtin_ctz(c); }

  inline int get_min(int l, int r) const { return comp(l, r) ? l : r; }

  explicit LinearRMQ(size_t n, const Comp &comp) : comp(comp) {
    vector<int> st;
    st.reserve(32);
    large.emplace_back();
    large.front().reserve(n / 32);
    small.reserve(n);
    for (int i = 0; i < n; i++) {
      while (!st.empty() && !comp(st.back(), i)) {
        st.pop_back();
      }
      small.emplace_back(st.empty() ? 0 : small[st.back()]);
      small.back() |= 1 << (i % 32);
      st.emplace_back(i);
      if ((i + 1) % 32 == 0) {
        large.front().emplace_back(st.front());
        st.clear();
      }
    }
    for (int i = 1; (i << 1) <= n / 32; i <<= 1) {
      vector<int> v;
      int csz = n / 32 + 1 - (i << 1);
      v.reserve(csz);
      for (int k = 0; k < csz; k++) {
        v.emplace_back(get_min(large.back()[k], large.back()[k + i]));
      }
      large.emplace_back(move(v));
    }
  }

  int fold(int l, int r) const {
    --r;
    int left = l / 32 + 1;
    int right = r / 32;
    if (left < right) {
      auto p = msb(right - left);
      return get_min(
          get_min((left - 1) * 32 + ctz(small[left * 32 - 1] & ~0 << l % 32),
                  large[p][left]),
          get_min(large[p][right - (1 << p)], right * 32 + ctz(small[r])));
    } else if (left == right) {
      return get_min((left - 1) * 32 + ctz(small[left * 32 - 1] & ~0 << l % 32),
                     left * 32 + ctz(small[r]));
    } else {
      return right * 32 + ctz(small[r] & ~0 << l % 32);
    }
  }
};

template <typename Comp>
LinearRMQ<Comp> get_linear_rmq(int n, const Comp &comp) {
  return LinearRMQ<Comp>(n, comp);
}

#line 1 "structure/others/linear-rmq.hpp"
/**
 * @brief Linear-RMQ
 **/
template <typename Comp>
struct LinearRMQ {
  vector<int> small;
  vector<vector<int> > large;

  LinearRMQ() = default;

  Comp comp;

  static inline int msb(int c) { return 31 - __builtin_clz(c); }

  static inline int ctz(int c) { return __builtin_ctz(c); }

  inline int get_min(int l, int r) const { return comp(l, r) ? l : r; }

  explicit LinearRMQ(size_t n, const Comp &comp) : comp(comp) {
    vector<int> st;
    st.reserve(32);
    large.emplace_back();
    large.front().reserve(n / 32);
    small.reserve(n);
    for (int i = 0; i < n; i++) {
      while (!st.empty() && !comp(st.back(), i)) {
        st.pop_back();
      }
      small.emplace_back(st.empty() ? 0 : small[st.back()]);
      small.back() |= 1 << (i % 32);
      st.emplace_back(i);
      if ((i + 1) % 32 == 0) {
        large.front().emplace_back(st.front());
        st.clear();
      }
    }
    for (int i = 1; (i << 1) <= n / 32; i <<= 1) {
      vector<int> v;
      int csz = n / 32 + 1 - (i << 1);
      v.reserve(csz);
      for (int k = 0; k < csz; k++) {
        v.emplace_back(get_min(large.back()[k], large.back()[k + i]));
      }
      large.emplace_back(move(v));
    }
  }

  int fold(int l, int r) const {
    --r;
    int left = l / 32 + 1;
    int right = r / 32;
    if (left < right) {
      auto p = msb(right - left);
      return get_min(
          get_min((left - 1) * 32 + ctz(small[left * 32 - 1] & ~0 << l % 32),
                  large[p][left]),
          get_min(large[p][right - (1 << p)], right * 32 + ctz(small[r])));
    } else if (left == right) {
      return get_min((left - 1) * 32 + ctz(small[left * 32 - 1] & ~0 << l % 32),
                     left * 32 + ctz(small[r]));
    } else {
      return right * 32 + ctz(small[r] & ~0 << l % 32);
    }
  }
};

template <typename Comp>
LinearRMQ<Comp> get_linear_rmq(int n, const Comp &comp) {
  return LinearRMQ<Comp>(n, comp);
}

Back to top page