#line 1 "a.cpp" #define PROBLEM "https://yukicoder.me/problems/no/924" #line 2 "/home/kuhaku/atcoder/github/algo/lib/template/template.hpp" #pragma GCC target("sse4.2,avx2,bmi2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include template bool chmax(T &a, const U &b) { return a < (T)b ? a = (T)b, true : false; } template bool chmin(T &a, const U &b) { return (T)b < a ? a = (T)b, true : false; } constexpr std::int64_t INF = 1000000000000000003; constexpr int Inf = 1000000003; constexpr int MOD = 1000000007; constexpr int MOD_N = 998244353; constexpr double EPS = 1e-7; constexpr double PI = M_PI; #line 3 "/home/kuhaku/atcoder/github/algo/lib/algorithm/compress.hpp" /** * @brief 座標圧縮 * * @tparam T 要素の型 */ template struct coordinate_compression { coordinate_compression() = default; coordinate_compression(const std::vector &_data) : data(_data) { build(); } const T &operator[](int i) const { return data[i]; } T &operator[](int i) { return data[i]; } void add(T x) { data.emplace_back(x); } void build() { std::sort(std::begin(data), std::end(data)); data.erase(std::unique(std::begin(data), std::end(data)), std::end(data)); } void build(const std::vector &v) { data = v; std::sort(std::begin(data), std::end(data)); data.erase(std::unique(std::begin(data), std::end(data)), std::end(data)); } bool exists(T x) const { auto it = std::lower_bound(std::begin(data), std::end(data), x); return it != std::end(data) && *it == x; } int get(T x) const { auto it = std::lower_bound(std::begin(data), std::end(data), x); return std::distance(std::begin(data), it); } int size() const { return std::size(data); } private: std::vector data; }; /** * @brief 座標圧縮 * * @tparam T 要素の型 * @param v * @return std::vector */ template std::vector compress(const std::vector &v) { coordinate_compression cps(v); std::vector res; for (auto &&x : v) res.emplace_back(cps.get(x)); return res; } #line 2 "/home/kuhaku/atcoder/github/algo/lib/binary_tree/fenwick_tree.hpp" /** * @brief フェニック木 * @see http://hos.ac/slides/20140319_bit.pdf * * @tparam T */ template struct fenwick_tree { fenwick_tree() : _size(), data() {} fenwick_tree(int n) : _size(n + 1), data(n + 1) {} fenwick_tree(const std::vector &v) : _size((int)v.size() + 1), data((int)v.size() + 1) { this->build(v); } template fenwick_tree(const std::vector &v) : _size((int)v.size() + 1), data((int)v.size() + 1) { this->build(v); } T operator[](int i) const { return this->sum(i + 1) - this->sum(i); } T at(int k) const { return this->operator[](k); } T get(int k) const { return this->operator[](k); } template void build(const std::vector &v) { for (int i = 0, n = v.size(); i < n; ++i) this->add(i, v[i]); } /** * @brief v[k] = val * * @param k index of array * @param val new value * @return void */ void update(int k, T val) { this->add(k, val - this->at(k)); } /** * @brief v[k] += val * * @param k index of array * @param val new value * @return void */ void add(int k, T val) { assert(0 <= k && k < this->_size); for (++k; k < this->_size; k += k & -k) this->data[k] += val; } /** * @brief chmax(v[k], val) * * @param k index of array * @param val new value * @return bool */ bool chmax(int k, T val) { if (this->at(k) >= val) return false; this->update(k, val); return true; } /** * @brief chmin(v[k], val) * * @param k index of value * @param val new value * @return bool */ bool chmin(int k, T val) { if (this->at(k) <= val) return false; this->update(k, val); return true; } /** * @brief v[0] + ... + v[n - 1] * * @return T */ T all_sum() const { return this->sum(this->_size); } /** * @brief v[0] + ... + v[k - 1] * * @param k index of array * @return T */ T sum(int k) const { assert(0 <= k && k <= this->_size); T res = 0; for (; k > 0; k -= k & -k) res += this->data[k]; return res; } /** * @brief v[a] + ... + v[b - 1] * * @param a first index of array * @param b last index of array * @return T */ T sum(int a, int b) const { return a < b ? this->sum(b) - this->sum(a) : 0; } /** * @brief binary search on fenwick_tree * * @param val target value * @return int */ int lower_bound(T val) const { if (val <= 0) return 0; int k = 1; while (k < this->_size) k <<= 1; int res = 0; for (; k > 0; k >>= 1) { if (res + k < this->_size && this->data[res + k] < val) val -= this->data[res += k]; } return res; } private: int _size; std::vector data; }; #line 2 "/home/kuhaku/atcoder/github/algo/lib/data_structure/bit_vector.hpp" /** * @brief 完備辞書 * * @see https://ei1333.github.io/library/structure/wavelet/succinct-indexable-dictionary.hpp */ struct bit_vector { bit_vector() = default; bit_vector(unsigned int _length) : length(_length), blocks((_length + 31) >> 5), bit((_length + 31) >> 5), sum((_length + 31) >> 5) {} void set(unsigned int k) { bit[k >> 5] |= 1U << (k & 31); } void build() { sum[0] = 0U; for (unsigned int i = 1; i < blocks; ++i) { sum[i] = sum[i - 1] + __builtin_popcount(bit[i - 1]); } } bool operator[](unsigned int k) const { return bit[k >> 5] >> (k & 31) & 1; } unsigned int rank(unsigned int k) const { return sum[k >> 5] + __builtin_popcount(bit[k >> 5] & ((1U << (k & 31)) - 1)); } unsigned int rank(bool val, unsigned int k) const { return val ? rank(k) : k - rank(k); } unsigned int select(unsigned int k) const { unsigned int sl = 0, sr = blocks + 1; while (sr - sl > 1) { unsigned int m = (sl + sr) >> 1; if (sum[m] < k) sl = m; else sr = m; } k -= sum[sl]; unsigned int bl = 0, br = 32; while (br - bl > 1) { unsigned int m = (bl + br) >> 1; if (__builtin_popcount(bit[sl] & ((1U << m) - 1)) < k) bl = m; else br = m; } return (sl << 5) + bl; } private: unsigned int length, blocks; std::vector bit, sum; }; #line 3 "/home/kuhaku/atcoder/github/algo/lib/data_structure/wavelet_matrix.hpp" /** * @brief ウェーブレット行列 * * @tparam T * @tparam L * * @see https://ei1333.github.io/library/structure/wavelet/wavelet-matrix.cpp.html */ template struct wavelet_matrix { wavelet_matrix() = default; wavelet_matrix(std::vector v) : length(v.size()) { std::vector l(length), r(length); for (int level = L - 1; level >= 0; --level) { matrix[level] = bit_vector(length + 1); int left = 0, right = 0; for (int i = 0; i < length; ++i) { if (v[i] >> level & 1) { matrix[level].set(i); r[right++] = v[i]; } else { l[left++] = v[i]; } } mid[level] = left; matrix[level].build(); v.swap(l); for (int i = 0; i < right; ++i) { v[left + i] = r[i]; } } } T access(int k) const { T res = 0; for (int level = L - 1; level >= 0; --level) { bool f = matrix[level][k]; if (f) res |= T(1) << level; k = matrix[level].rank(f, k) + mid[level] * f; } return res; } T operator[](int k) const { return access(k); } /** * @brief count i s.t. (0 <= i < r) && v[i] == x * * @param x * @param r * @return int */ int rank(int r, T x) const { int l = 0; for (int level = L - 1; level >= 0; --level) { std::tie(l, r) = succ((x >> level) & 1, l, r, level); } return r - l; } /** * @brief count i s.t. (l <= i < r) && v[i] == x * * @param l * @param r * @param x * @return int */ int rank(int l, int r, T x) const { return rank(r, x) - rank(l, x); } /** * @brief k-th smallest number in v[l ... r-1] * * @param l * @param r * @param k * @return T */ T kth_smallest(int l, int r, int k) const { assert(0 <= k && k < r - l); T res = 0; for (int level = L - 1; level >= 0; --level) { int cnt = matrix[level].rank(false, r) - matrix[level].rank(false, l); bool f = cnt <= k; if (f) { res |= T(1) << level; k -= cnt; } std::tie(l, r) = succ(f, l, r, level); } return res; } /** * @brief k-th largest number in v[l ... r-1] * * @param l * @param r * @param k * @return T */ T kth_largest(int l, int r, int k) const { return kth_smallest(l, r, r - l - k - 1); } /** * @brief count i s.t. (l <= i < r) && (v[i] < upper) * * @param l * @param r * @param upper * @return int */ int range_freq(int l, int r, T upper) const { int res = 0; for (int level = L - 1; level >= 0; --level) { bool f = ((upper >> level) & 1); if (f) res += matrix[level].rank(false, r) - matrix[level].rank(false, l); std::tie(l, r) = succ(f, l, r, level); } return res; } /** * @brief count i s.t. (l <= i < r) && (lower <= v[i] < upper) * * @param l * @param r * @param lower * @param upper * @return int */ int range_freq(int l, int r, T lower, T upper) const { return range_freq(l, r, upper) - range_freq(l, r, lower); } /** * @brief max v[i] s.t. (l <= i < r) && (v[i] < upper) * * @param l * @param r * @param upper * @return T */ T prev_value(int l, int r, T upper) const { int cnt = range_freq(l, r, upper); return cnt == 0 ? T(-1) : kth_smallest(l, r, cnt - 1); } /** * @brief min v[i] s.t. (l <= i < r) && (lower <= v[i]) * * @param l * @param r * @param lower * @return T */ T next_value(int l, int r, T lower) const { int cnt = range_freq(l, r, lower); return cnt == r - l ? T(-1) : kth_smallest(l, r, cnt); } private: int length; bit_vector matrix[L]; int mid[L]; std::pair succ(bool f, int l, int r, int level) const { return {matrix[level].rank(f, l) + mid[level] * f, matrix[level].rank(f, r) + mid[level] * f}; } }; #line 4 "/home/kuhaku/atcoder/github/algo/lib/data_structure/compressed_wavelet_matrix.hpp" /** * @brief ウェーブレット行列 * * @tparam T * @tparam L * * @see https://ei1333.github.io/library/structure/wavelet/wavelet-matrix.cpp.html */ template struct compressed_wavelet_matrix { compressed_wavelet_matrix() = default; compressed_wavelet_matrix(const std::vector &v) : cps(v) { int n = v.size(); std::vector t(n); for (int i = 0; i < n; ++i) t[i] = cps.get(v[i]); mat = wavelet_matrix(t); } T access(int k) const { return cps[mat.access(k)]; } T operator[](int k) const { return access(k); } /** * @brief count i s.t. (0 <= i < r) && v[i] == x * * @param x * @param r * @return int */ int rank(int r, T x) const { auto pos = cps.get(x); if (pos == cps.size() || cps[pos] != x) return 0; return mat.rank(r, pos); } /** * @brief count i s.t. (l <= i < r) && v[i] == x * * @param l * @param r * @param x * @return int */ int rank(int l, int r, T x) const { return rank(r, x) - rank(l, x); } /** * @brief k-th smallest number in v[l ... r-1] * * @param l * @param r * @param k * @return T */ T kth_smallest(int l, int r, int k) const { return cps[mat.kth_smallest(l, r, k)]; } /** * @brief k-th largest number in v[l ... r-1] * * @param l * @param r * @param k * @return T */ T kth_largest(int l, int r, int k) const { return cps[mat.kth_largest(l, r, k)]; } /** * @brief count i s.t. (l <= i < r) && (v[i] < upper) * * @param l * @param r * @param upper * @return int */ int range_freq(int l, int r, T upper) const { return mat.range_freq(l, r, cps.get(upper)); } /** * @brief count i s.t. (l <= i < r) && (lower <= v[i] < upper) * * @param l * @param r * @param lower * @param upper * @return int */ int range_freq(int l, int r, T lower, T upper) const { return mat.range_freq(l, r, cps.get(lower), cps.get(upper)); } /** * @brief max v[i] s.t. (l <= i < r) && (v[i] < upper) * * @param l * @param r * @param upper * @return T */ T prev_value(int l, int r, T upper) const { auto res = mat.prev_value(l, r, cps.get(upper)); return res == -1 ? T(-1) : cps[res]; } /** * @brief min v[i] s.t. (l <= i < r) && (lower <= v[i]) * * @param l * @param r * @param lower * @return T */ T next_value(int l, int r, T lower) const { auto res = mat.next_value(l, r, cps.get(lower)); return res == -1 ? T(-1) : cps[res]; } private: wavelet_matrix mat; coordinate_compression cps; }; #line 3 "/home/kuhaku/atcoder/github/algo/lib/template/macro.hpp" #define FOR(i, m, n) for (int i = (m); i < int(n); ++i) #define FORR(i, m, n) for (int i = (m)-1; i >= int(n); --i) #define FORL(i, m, n) for (int64_t i = (m); i < int64_t(n); ++i) #define rep(i, n) FOR (i, 0, n) #define repn(i, n) FOR (i, 1, n + 1) #define repr(i, n) FORR (i, n, 0) #define repnr(i, n) FORR (i, n + 1, 1) #define all(s) (s).begin(), (s).end() #line 3 "/home/kuhaku/atcoder/github/algo/lib/template/sonic.hpp" struct Sonic { Sonic() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); } constexpr void operator()() const {} } sonic; #line 5 "/home/kuhaku/atcoder/github/algo/lib/template/atcoder.hpp" using namespace std; using ll = std::int64_t; using ld = long double; template std::istream &operator>>(std::istream &is, std::pair &p) { return is >> p.first >> p.second; } template std::istream &operator>>(std::istream &is, std::vector &v) { for (T &i : v) is >> i; return is; } template std::ostream &operator<<(std::ostream &os, const std::pair &p) { return os << '(' << p.first << ',' << p.second << ')'; } template std::ostream &operator<<(std::ostream &os, const std::vector &v) { for (auto it = v.begin(); it != v.end(); ++it) { os << (it == v.begin() ? "" : " ") << *it; } return os; } template void co(Head &&head, Tail &&...tail) { if constexpr (sizeof...(tail) == 0) std::cout << head << '\n'; else std::cout << head << ' ', co(std::forward(tail)...); } template void ce(Head &&head, Tail &&...tail) { if constexpr (sizeof...(tail) == 0) std::cerr << head << '\n'; else std::cerr << head << ' ', ce(std::forward(tail)...); } template auto make_vector(T x, int arg, Args... args) { if constexpr (sizeof...(args) == 0) return std::vector(arg, x); else return std::vector(arg, make_vector(x, args...)); } void setp(int n) { std::cout << std::fixed << std::setprecision(n); } void Yes(bool is_correct = true) { std::cout << (is_correct ? "Yes" : "No") << '\n'; } void No(bool is_not_correct = true) { Yes(!is_not_correct); } void YES(bool is_correct = true) { std::cout << (is_correct ? "YES" : "NO") << '\n'; } void NO(bool is_not_correct = true) { YES(!is_not_correct); } void Takahashi(bool is_correct = true) { std::cout << (is_correct ? "Takahashi" : "Aoki") << '\n'; } void Aoki(bool is_not_correct = true) { Takahashi(!is_not_correct); } #line 6 "a.cpp" int main(void) { int n, q; cin >> n >> q; vector a(n); cin >> a; vector> b(q); cin >> b; compressed_wavelet_matrix wm(a); coordinate_compression cc(a); int m = cc.size(); auto c = compress(a); vector ok(q); rep (i, q) { ok[i] = wm.kth_smallest(b[i].first - 1, b[i].second, (b[i].second - b[i].first + 1) / 2); } vector> query(n + 1); rep (i, q) { auto [l, r] = b[i]; query[l - 1].emplace_back(~i); query[r].emplace_back(i); } vector ans(q); fenwick_tree ft1(m), ft2(m); rep (i, n) { ft1.add(c[i], a[i]); ft2.add(c[i], 1); for (auto &&idx : query[i + 1]) { if (idx >= 0) { int t = cc.get(ok[idx]); ans[idx] += ft1.sum(t, m); ans[idx] -= ok[idx] * ft2.sum(t, m); ans[idx] += ok[idx] * ft2.sum(t); ans[idx] -= ft1.sum(t); } else { idx = ~idx; int t = cc.get(ok[idx]); ans[idx] -= ft1.sum(t, m); ans[idx] += ok[idx] * ft2.sum(t, m); ans[idx] -= ok[idx] * ft2.sum(t); ans[idx] += ft1.sum(t); } } } for (auto x : ans) co(x); return 0; }