結果

問題 No.2901 Logical Sum of Substring
コンテスト
ユーザー hitonanode
提出日時 2024-09-21 22:19:53
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 349 ms / 3,000 ms
コード長 1,923 bytes
コンパイル時間 1,863 ms
コンパイル使用メモリ 130,084 KB
実行使用メモリ 88,808 KB
最終ジャッジ日時 2024-09-21 22:20:06
合計ジャッジ時間 11,608 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <utility>
#include <vector>
using namespace std;
template <typename T> bool chmax(T &m, const T q) { return m < q ? (m = q, true) : false; }
template <typename T> bool chmin(T &m, const T q) { return m > q ? (m = q, true) : false; }

#include <atcoder/segtree>

constexpr int inf = 1 << 28;
int K;
int full;

struct S {
    int len = 0;
    int ans = inf;

    vector<pair<int, int>> left_cs, right_cs;

    static S gen(int a) { return S{1, (a == full ? 1 : inf), {{1, a}}, {{1, a}}}; }

    static vector<pair<int, int>> merge(const vector<pair<int, int>> &v1, const vector<pair<int, int>> &v2, int len1) {
        auto ret = v1;
        int mask = v1.empty() ? 0 : v1.back().second;
        for (auto [k, v] : v2) {
            if (chmax(mask, mask | v)) ret.emplace_back(k + len1, mask);
        }

        return ret;
    }
};

S op(S l, S r) {
    auto len = l.len + r.len;
    auto left_cs = S::merge(l.left_cs, r.left_cs, l.len);
    auto right_cs = S::merge(r.right_cs, l.right_cs, r.len);
    int ans = min(l.ans, r.ans);

    for (auto [k1, v1] : l.right_cs) {
        for (auto [k2, v2] : r.left_cs) {
            if ((v1 | v2) == full) {
                chmin(ans, k1 + k2);
                break;
            }
        }
    }

    return S{len, ans, move(left_cs), move(right_cs)};
}

S e() { return S(); }

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

    int N;
    cin >> N >> K;
    full = (1 << K) - 1;

    vector<S> init;
    while (N--) {
        int a;
        cin >> a;
        init.emplace_back(S::gen(a));
    }

    int Q;
    cin >> Q;
    atcoder::segtree<S, op, e> seg(init);

    while (Q--) {
        int tp, l, r;
        cin >> tp >> l >> r;
        --l;

        if (tp == 1) {
            seg.set(l, S::gen(r));
        } else {
            auto p = seg.prod(l, r).ans;
            cout << (p == inf ? -1 : p) << '\n';
        }
    }
}
0