結果

問題 No.3078 Difference Sum Query
コンテスト
ユーザー V_Melville
提出日時 2025-03-29 18:53:16
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 664 ms / 2,000 ms
コード長 1,818 bytes
コンパイル時間 3,555 ms
コンパイル使用メモリ 289,744 KB
実行使用メモリ 53,700 KB
最終ジャッジ日時 2025-03-29 18:53:38
合計ジャッジ時間 19,898 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)

using namespace std;
using ll = long long;

struct segtree {
    int n;
    vector<vector<ll>> a;
    vector<vector<ll>> s;
    segtree(int mx) {
        n = 1;
        while (n < mx) n <<= 1;
        a.resize(n*2);
        s.resize(n*2, vector<ll>(1));
    }
    void set(int i, ll x) {
        a[n+i] = {x};
        s[n+i].push_back(x);
    }
    void init() {
        for (int i = n-1; i >= 1; --i) {
            int l = i<<1, r = l|1;
            merge(a[l].begin(), a[l].end(), a[r].begin(), a[r].end(), back_inserter(a[i]));
            s[i] = vector<ll>(a[i].size()+1);
            rep(j, a[i].size()) s[i][j+1] = s[i][j]+a[i][j];
        }
    }
    
    pair<int, ll> get(int i, ll x) {
        int n = a[i].size();
        int j = lower_bound(a[i].begin(), a[i].end(), x) - a[i].begin();
        return {(n-j)-j, (s[i][n]-s[i][j])-(s[i][j]-s[i][0])};
    }
    
    pair<int, ll> solve(int l, int r, ll x) {
        l += n; r += n;
        int num = 0; ll sum = 0;
        while (l < r) {
            if (l&1) {
                auto [tnum, tsum] = get(l, x);
                num += tnum, sum += tsum;
                l++;
            }
            if (r&1) {
                r--;
                auto [tnum, tsum] = get(r, x);
                num += tnum, sum += tsum;
            }
            l >>= 1; r >>= 1;
        }
        return {num, sum};
    }
};

int main() {
    int n, q;
    cin >> n >> q;
    
    segtree t(n);
    rep(i, n) {
        ll a;
        cin >> a;
        t.set(i, a);
    }
    t.init();
    
    rep(qi, q) {
        int l, r; ll x;
        cin >> l >> r >> x;
        --l;
        auto [num, sum] = t.solve(l, r, x);
        ll ans = sum - num*x;
        cout << ans << '\n';
    }
    
    return 0;
}
0