test/yosupo/data_structure/static_range_inversions.test.cpp
✅
AC
· 最終更新 2026-09-07
· GitHub
問題: https://judge.yosupo.jp/problem/static_range_inversions_query
依存するヘッダ(3)
実装
// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/static_range_inversions_query
#include <algorithm>
#include <cstdint>
#include <iostream>
#include <vector>
#include "algorithm/coordinate_compression.hpp"
#include "algorithm/mo.hpp"
#include "data_structure/fenwick_tree.hpp"
int main(void) {
int n, q;
std::cin >> n >> q;
std::vector<int> a(n);
for (auto &e : a) std::cin >> e;
a = compress(a);
std::int64_t sum = 0;
int size = *max_element(a.begin(), a.end()) + 1;
FenwickTree<int> ft(size);
auto fl = [&](int idx) {
sum += ft.sum(a[idx]);
ft.add(a[idx], 1);
};
auto fr = [&](int idx) {
sum += ft.sum(a[idx] + 1, size);
ft.add(a[idx], 1);
};
auto gl = [&](int idx) {
sum -= ft.sum(a[idx]);
ft.add(a[idx], -1);
};
auto gr = [&](int idx) {
sum -= ft.sum(a[idx] + 1, size);
ft.add(a[idx], -1);
};
std::vector<std::int64_t> ans(q);
auto rem = [&](int idx) { return ans[idx] = sum; };
Mo mo(n);
mo.input(q, 0);
mo.solve(fl, fr, gl, gr, rem);
for (auto x : ans) std::cout << x << '\n';
return 0;
}