結果

問題 No.2665 Minimize Inversions of Deque
ユーザー suisen
提出日時 2023-12-28 21:22:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 1,419 bytes
コンパイル時間 940 ms
コンパイル使用メモリ 82,476 KB
最終ジャッジ日時 2025-02-18 15:00:17
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <deque>
#include <iostream>
#include <vector>

#include <atcoder/fenwicktree>

std::pair<long long, std::vector<int>> solve(int n, const std::vector<int>& p) {
    long long inv = 0;
    std::deque<int> a;

    atcoder::fenwick_tree<int> ft(n);
    for (int i = 0; i < n; ++i) {
        const int cnt_lt = ft.sum(0, p[i]);
        const int cnt_gt = i - cnt_lt;
        ft.add(p[i], 1);

        const bool ok_l = cnt_lt <= cnt_gt;
        const bool ok_r = cnt_lt >= cnt_gt;
        if (ok_l and ok_r) {
            if (a.empty() or a.front() < p[i]) {
                a.push_back(p[i]);
            } else {
                a.push_front(p[i]);
            }
        } else if (ok_l) {
            a.push_front(p[i]);
        } else if (ok_r) {
            a.push_back(p[i]);
        } else {
            assert(false);
        }
        inv += std::min(cnt_lt, cnt_gt);
    }
    return { inv, std::vector<int>(a.begin(), a.end()) };
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int t;
    std::cin >> t;
    while (t--) {
        int n;
        std::cin >> n;

        std::vector<int> p(n);
        for (auto&& e : p) std::cin >> e, --e;

        auto [inv, a] = solve(n, p);

        std::cout << inv << '\n';
        for (int i = 0; i < n; ++i) {
            if (i) std::cout << ' ';
            std::cout << a[i] + 1;
        }
        std::cout << '\n';
    }
}
0