#636147 (C++17) No.1441 MErGe

• 問題: No.1441 MErGe
• ユーザー: KoD
• 提出日時: 2021-03-31 11:03:24
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 193 ms / 1,000 ms
• コード長: 3,280 bytes
• コンパイル時間: 2,348 ms
• コンパイル使用メモリ: 205,852 KB
• 実行使用メモリ: 7,936 KB
• 最終ジャッジ日時: 2024-05-08 10:41:46
• 合計ジャッジ時間: 8,471 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	3 ms; 5,376 KB
testcase_04	AC	3 ms; 5,376 KB
testcase_05	AC	5 ms; 5,376 KB
testcase_06	AC	5 ms; 5,376 KB
testcase_07	AC	5 ms; 5,376 KB
testcase_08	AC	36 ms; 5,376 KB
testcase_09	AC	35 ms; 5,376 KB
testcase_10	AC	52 ms; 5,376 KB
testcase_11	AC	48 ms; 5,376 KB
testcase_12	AC	55 ms; 5,376 KB
testcase_13	AC	141 ms; 7,680 KB
testcase_14	AC	154 ms; 7,680 KB
testcase_15	AC	154 ms; 7,552 KB
testcase_16	AC	144 ms; 7,680 KB
testcase_17	AC	146 ms; 7,552 KB
testcase_18	AC	120 ms; 6,528 KB
testcase_19	AC	135 ms; 7,168 KB
testcase_20	AC	108 ms; 6,272 KB
testcase_21	AC	89 ms; 6,016 KB
testcase_22	AC	120 ms; 5,760 KB
testcase_23	AC	189 ms; 7,808 KB
testcase_24	AC	193 ms; 7,936 KB
testcase_25	AC	189 ms; 7,936 KB
testcase_26	AC	191 ms; 7,936 KB
testcase_27	AC	190 ms; 7,808 KB
testcase_28	AC	103 ms; 7,808 KB
testcase_29	AC	104 ms; 7,936 KB

ソースコード

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