結果

問題 No.1441 MErGe
ユーザー KoD
提出日時 2021-03-31 11:03:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 168 ms / 1,000 ms
コード長 3,280 bytes
コンパイル時間 2,037 ms
コンパイル使用メモリ 197,264 KB
最終ジャッジ日時 2025-01-20 01:11:25
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using i32 = std::int32_t;
using u32 = std::uint32_t;
using i64 = std::int64_t;
using u64 = std::uint64_t;
using i128 = __int128_t;
using u128 = __uint128_t;
using isize = std::ptrdiff_t;
using usize = std::size_t;

class rep {
    struct Iter {
        usize itr;
        constexpr Iter(const usize pos) noexcept: itr(pos) { }
        constexpr void operator ++ () noexcept { ++itr; }
        constexpr bool operator != (const Iter& other) const noexcept { return itr != other.itr; }
        constexpr usize operator * () const noexcept { return itr; }
    };
    const Iter first, last;
public:
    explicit constexpr rep(const usize first, const usize last) noexcept: first(first), last(std::max(first, last)) { }
    constexpr Iter begin() const noexcept { return first; }
    constexpr Iter end() const noexcept { return last; }
};

constexpr u64 ceil_log2(const u64 x) {
    u64 e = 0;
    while (((u64) 1 << e) < x) ++e;
    return e;
}

template <class T>
class FenwickTree {
    usize logn;
    std::vector<T> data;
public:
    explicit FenwickTree(const usize size = 0) {
        logn = ceil_log2(size + 1) - 1;
        data = std::vector<T>(size + 1, T{});
    }

    usize size() const { return data.size() - 1; }

    void add(usize i, const T& x) { 
        assert(i < size());
        i += 1;
        while (i < data.size()) {
            data[i] += x;
            i += i & -i;
        }
    }
    void subtract(usize i, const T& x) { 
        assert(i < size());
        i += 1;
        while (i < data.size()) {
            data[i] -= x;
            i += i & -i;
        }
    }

    T fold(usize l, usize r) const {
        assert(l <= r and r <= size());
        T ret{};
        while (l < r) {
            ret += data[r];
            r -= r & -r;
        }
        while (r < l) {
            ret -= data[l];
            l -= l & -l;
        }
        return ret;
    }

    template <class F>
    usize max_right(const F &f) const {
        assert(f(T{}));
        usize i = 0;
        T sum{};
        for (usize k = (1 << logn); k > 0; k >>= 1) {
            if (i + k <= size() && f(sum + data[i + k])) {
                i += k;
                sum += data[i];
            }
        }
        return i;
    }
};

template <class T>
using Vec = std::vector<T>;

void Y1441_main() {
    usize N, Q;
    std::cin >> N >> Q;
    Vec<i64> A(N);
    for (auto &x: A) {
        std::cin >> x;
    }
    Vec<i64> sum(N + 1);
    for (auto i: rep(0, N)) {
        sum[i + 1] = sum[i] + A[i];
    }
    FenwickTree<usize> fen(N + 1);
    for (auto i: rep(0, N + 1)) {
        fen.add(i, 1);
    }
    while (Q--) {
        usize t, l, r;
        std::cin >> t >> l >> r;
        l -= 1;
        if (t == 1) {
            for (auto _: rep(l + 1, r)) {
                fen.subtract(fen.max_right([&](const usize x) { return x <= l + 1; }), 1);
            }
        }
        else {
            l = fen.max_right([&](const usize x) { return x <= l; });
            r = fen.max_right([&](const usize x) { return x <= r; });
            std::cout << sum[r] - sum[l] << '\n';
        }
    }
    return;
}

int main() {
    std::ios_base::sync_with_stdio(false);
#ifndef LOCAL
    std::cin.tie(nullptr);
#endif
    Y1441_main();
    return 0;
}
0