結果

問題 No.877 Range ReLU Query
ユーザー veqcc
提出日時 2019-09-09 20:07:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 136 ms / 2,000 ms
コード長 2,482 bytes
コンパイル時間 1,370 ms
コンパイル使用メモリ 118,756 KB
実行使用メモリ 15,104 KB
最終ジャッジ日時 2024-11-08 10:20:07
合計ジャッジ時間 4,163 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <functional>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstring>
#include <string>
#include <vector>
#include <random>
#include <bitset>
#include <queue>
#include <cmath>
#include <stack>
#include <set>
#include <map>
typedef long long ll;
using namespace std;
const ll MOD = 1000000007LL;

template <typename T>
class SegmentTree {
    using F = function<T(T, T)>;
    int n;
    F f;
    T t;
    vector <T> dat;
public:
    SegmentTree(F f, T t) : f(f), t(t) {}
    void init(int n_) {
        n = 1;
        while (n < n_) n <<= 1;
        dat.assign(n << 1, t);
    }
    void build(const vector <T> &v) {
        auto n_ = (int)v.size();
        init(n_);
        for (int i = 0; i < n_; i++) dat[n + i] = v[i];
        for (int i = n - 1; i > 0; i--) dat[i] = f(dat[(i << 1) | 0], dat[(i << 1) | 1]);
    }
    void set_val(int k, T x) {
        dat[k += n] = x;
        while (k >>= 1) dat[k] = f(dat[(k << 1) | 0], dat[(k << 1) | 1]);
    }
    T query(int a, int b) {
        T vl = t, vr = t;
        for (int l = a + n, r = b + n; l < r; l >>= 1, r >>= 1) {
            if (l & 1) vl = f(vl, dat[l++]);
            if (r & 1) vr = f(dat[--r], vr);
        }
        return f(vl, vr);
    }
};

typedef pair <ll, ll> P;

struct Query {
    int l, r;
    ll x;
};

struct Val {
    ll val;
    int type, idx;
    bool operator>(const Val &a) const {
        return val > a.val;
    }
};

int main() {
    cin.sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int n, q, c;
    cin >> n >> q;

    vector <ll> a(n);
    for (int i = 0; i < n; i++) cin >> a[i];

    vector <Query> Q(q);
    for (int i = 0; i < q; i++) cin >> c >> Q[i].l >> Q[i].r >> Q[i].x;

    vector <Val> v(n + q);
    for (int i = 0; i < n; i++) v[i] = {a[i], 0, i};
    for (int i = 0; i < q; i++) v[n + i] = {Q[i].x, 1, i};

    sort(v.begin(), v.end(), greater<Val>());

    vector <ll> ans(q);

    auto f = [](P a, P b) {
        return P(a.first + b.first, a.second + b.second);
    };

    SegmentTree<P> seg(f, P(0, 0));
    seg.build(vector<P>(n, P(0, 0)));

    for (int i = 0; i < n + q; i++) {
        if (v[i].type == 0) {
            seg.set_val(v[i].idx, {v[i].val, 1});
        } else {
            int idx = v[i].idx;
            P ret = seg.query(Q[idx].l - 1, Q[idx].r);
            ans[idx] = ret.first - v[i].val * ret.second;
        }
    }

    for (int i = 0; i < q; i++) {
        cout << ans[i] << '\n';
    }

    return 0;
}
0