結果

問題 No.2223 Merged Med
ユーザー 👑 hitonanodehitonanode
提出日時 2023-03-11 14:25:51
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 343 ms / 7,000 ms
コード長 2,381 bytes
コンパイル時間 2,260 ms
コンパイル使用メモリ 140,600 KB
実行使用メモリ 18,588 KB
最終ジャッジ日時 2023-10-18 09:52:24
合計ジャッジ時間 8,335 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 2 ms
4,348 KB
testcase_09 AC 2 ms
4,348 KB
testcase_10 AC 2 ms
4,348 KB
testcase_11 AC 2 ms
4,348 KB
testcase_12 AC 2 ms
4,348 KB
testcase_13 AC 106 ms
8,564 KB
testcase_14 AC 55 ms
6,088 KB
testcase_15 AC 146 ms
13,776 KB
testcase_16 AC 280 ms
16,660 KB
testcase_17 AC 73 ms
9,156 KB
testcase_18 AC 50 ms
5,920 KB
testcase_19 AC 95 ms
8,300 KB
testcase_20 AC 323 ms
18,280 KB
testcase_21 AC 324 ms
18,336 KB
testcase_22 AC 340 ms
18,284 KB
testcase_23 AC 325 ms
18,016 KB
testcase_24 AC 343 ms
18,280 KB
testcase_25 AC 336 ms
18,280 KB
testcase_26 AC 322 ms
18,340 KB
testcase_27 AC 308 ms
18,020 KB
testcase_28 AC 334 ms
18,016 KB
testcase_29 AC 339 ms
18,280 KB
testcase_30 AC 82 ms
18,588 KB
testcase_31 AC 75 ms
17,220 KB
testcase_32 AC 11 ms
4,348 KB
testcase_33 AC 216 ms
18,016 KB
testcase_34 AC 135 ms
18,316 KB
testcase_35 AC 130 ms
17,220 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <iostream>
#include <utility>
#include <vector>
using namespace std;
#define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i<i##_end_;i++)
#define REP(i, n) FOR(i,0,n)

#include <atcoder/segtree>

struct S {
    int sum = 0;
    int min_cumsum = 0;
    int max_cumsum = 0;
    int diff_max = 0;
    static S plus() { return S{1, 0, 1, 1}; }
    static S minus() { return S{-1, -1, 0, 0}; }
};

S e() { return S(); }
S op(S l, S r) {
    return S{
        l.sum + r.sum,
        std::min(l.min_cumsum, l.sum + r.min_cumsum),
        std::max(l.max_cumsum, l.sum + r.max_cumsum),
        std::max({l.diff_max, r.diff_max, l.sum + r.max_cumsum - l.min_cumsum}),
    };
}

int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);
    int N, Q;
    cin >> N >> Q;
    vector<int> A(N);
    for (auto &x : A) cin >> x;

    vector<pair<int, int>> query(Q);
    for (auto &[l, r] : query) cin >> l >> r, --l;

    vector<int> bin_ng(Q, 0), bin_ok(Q, N);

    vector<vector<int>> a2is(N + 1), c2qs(N + 1);
    REP(i, N) a2is.at(A.at(i)).push_back(i);

    vector<S> init(N, S::plus());
    atcoder::segtree<S, op, e> tree;
    while (true) {
        tree = atcoder::segtree<S, op, e>(init);

        bool flg = false;
        REP(q, Q) {
            if (bin_ng.at(q) + 1 < bin_ok.at(q)) {
                c2qs.at((bin_ng.at(q) + bin_ok.at(q)) / 2).emplace_back(q);
                flg = true;
            }
        }
        if (!flg) break;

        int added = 0;

        REP(c, c2qs.size()) {
            for (const int q : c2qs.at(c)) {
                while (added < c) {
                    ++added;
                    for (int i : a2is.at(added)) tree.set(i, S::minus());
                }
                const auto [l, r] = query.at(q);
                const auto p = tree.prod(l, r);

                bool success = false;
                int reduce_plus = 0;
                if (p.diff_max > 1) reduce_plus = p.diff_max - 1;
                int cnt_minus = (r - l - p.sum) / 2;
                int num_plus = r - l - cnt_minus - reduce_plus;
                if (cnt_minus > num_plus) success = true;

                (success ? bin_ok : bin_ng).at(q) = c;
            }
            c2qs.at(c).clear();
        }
    }
    for (auto x : bin_ok) cout << x << '\n';
}
0