結果

問題 No.876 Range Compress Query
ユーザー ooaiu
提出日時 2024-11-14 12:57:59
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 128 ms / 2,000 ms
コード長 1,312 bytes
コンパイル時間 3,854 ms
コンパイル使用メモリ 281,124 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-14 12:58:07
合計ジャッジ時間 6,649 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

#ifndef LOCAL
#include <bits/stdc++.h>

using namespace std;

#define debug(...) (void(0))
#else
#include "algo/debug.h"
#endif

#include <atcoder/segtree>
int op(int a, int b) {
    return a + b;
}
int e() {
    return 0;
}
void solve() {
    int N, Q;
    cin >> N >> Q;
    vector<int> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];
    vector<long long> diff(N);
    for (int i = 1; i < N; i++) diff[i] = A[i] - A[i - 1];
    vector<int> B(N);
    for (int i = 1; i < N; i++) {
        B[i] = diff[i] != 0;
    }
    atcoder::segtree<int, op, e> seg(B);

    auto Update = [&](int i) -> void {
        if (diff[i] == 0)
            seg.set(i, 0);
        else
            seg.set(i, 1);
    };
    while (Q--) {
        int t;
        cin >> t;
        int l, r;
        cin >> l >> r;
        if (t == 1) {
            int x;
            cin >> x;
            l--;
            if (l != 0) {
                diff[l] += x;
                Update(l);
            }
            if (r != N) {
                diff[r] -= x;
                Update(r);
            }
        } else {
            cout << seg.prod(l, r) + 1 << endl;
        }
    }
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int tt = 1;
    // std::cin >> tt;
    while (tt--) {
        solve();
    }
}
0