結果

問題 No.3411 Range Clamp Sum
コンテスト
ユーザー AwashAmityOak
提出日時 2025-11-18 23:14:17
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
(最新)
TLE  
(最初)
実行時間 5,407 ms / 10,000 ms
コード長 2,535 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 8,481 ms
コンパイル使用メモリ 336,340 KB
実行使用メモリ 19,316 KB
最終ジャッジ日時 2025-12-17 23:31:23
合計ジャッジ時間 76,183 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

#define int long long
signed main() {
    int N, Q;
    cin >> N >> Q;
    vector<int> A(N);
    rep(i, N) cin >> A[i];
    
    int B = (int)(sqrtl(N) + 0.5);
    vector<vector<tuple<int, int, int, int, int>>> queries((Q + B - 1) / B);
    rep(q, Q) {
        int t;
        cin >> t;
        
        if (t == 1) {
            int x, y;
            cin >> x >> y;
            --x;
            
            queries[q / B].emplace_back(t, x, y, -1, -1);
        } else {
            int l, r, a, b;
            cin >> l >> r >> a >> b;
            --l;
            
            queries[q / B].emplace_back(t, l, r, a, b);
        }
    }
    
    for (auto& block : queries) {
        vector<int> ans(block.size());
        
        vector<tuple<int, int, int, int, int>> events;
        rep(i, N) events.emplace_back(A[i], i, -1, -1, -1);
        rep(q, block.size()) {
            auto [t, l, r, a, b] = block[q];
            if (t == 1) continue;
            
            events.emplace_back(a, l, r, q, 0);
            events.emplace_back(b, l, r, q, 1);
        }
        sort(events.begin(), events.end());
        
        fenwick_tree<int> sum0(N), sum1(N);
        for (auto [x, l, r, q, t] : events) {
            if (t == -1) {
                sum0.add(l, 1);
                sum1.add(l, x);
            } else if (t == 0) {
                ans[q] -= sum1.sum(l, r);
                ans[q] += x * sum0.sum(l, r);
            } else {
                ans[q] += sum1.sum(l, r);
                ans[q] += x * (r - l - sum0.sum(l, r));
            }
        }
        
        map<int, int> update_history;
        rep(q, block.size()) {
            auto [t, l, r, a, b] = block[q];
            if (t == 1) {
                int x = l;
                int y = r;
                
                update_history[x] = y;
            } else {
                if (a > b) {
                    cout << a * (r - l) << '\n';
                    continue;
                }
                
                for (auto [i, x] : update_history) {
                    if (l <= i && i < r) {
                        ans[q] -= clamp(A[i], a, b);
                        ans[q] += clamp(x, a, b);
                    }
                }
                cout << ans[q] << '\n';
            }
        }
        
        for (auto [i, x] : update_history) {
            A[i] = x;
        }
    }
}
0