結果

問題 No.877 Range ReLU Query
ユーザー Tatsu_mr
提出日時 2024-06-13 23:10:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 341 ms / 2,000 ms
コード長 1,098 bytes
コンパイル時間 4,712 ms
コンパイル使用メモリ 264,164 KB
最終ジャッジ日時 2025-02-21 21:35:49
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;

using lint = long long;

struct S {
    lint val, size;
};

S op(S a, S b) {
    return {a.val + b.val, a.size + b.size};
}

S e() {
    return {0LL, 0LL};
}

int main() {
    int n, q;
    cin >> n >> q;
    /*
    更新クエリ
    {x, i, -1, -1}
    取得クエリ
    {x, l, r, i}
    */
    vector<tuple<lint, lint, lint, lint>> tup;
    for (int i = 0; i < n; i++) {
        lint a;
        cin >> a;
        tup.push_back({a, i, -1, -1});
    }
    for (int i = 0; i < q; i++) {
        int t, l, r;
        lint x;
        cin >> t >> l >> r >> x;
        l--;
        r--;
        tup.push_back({x, l, r, i});
    }
    sort(tup.rbegin(), tup.rend());
    segtree<S, op, e> seg(n);
    vector<lint> ans(q);
    for (auto [x, l, r, i] : tup) {
        if (i == -1) {
            seg.set(l, {x, 1LL});
        } else {
            S res = seg.prod(l, r + 1);
            ans[i] = res.val - res.size * x;
        }
    }
    for (int i = 0; i < q; i++) {
        cout << ans[i] << endl;
    }
}
0