algo — C++23 日本語リファレンス GitHub

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;
}