結果

問題 No.1099 Range Square Sum
コンテスト
ユーザー tokusakurai
提出日時 2020-06-27 09:28:06
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 399 ms / 2,000 ms
コード長 3,392 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,901 ms
コンパイル使用メモリ 172,472 KB
実行使用メモリ 17,136 KB
最終ジャッジ日時 2024-07-05 08:53:59
合計ジャッジ時間 7,315 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i = 0; i < n; i++)
#define rep2(i, x, n) for(int i = x; i <= n; i++)
#define rep3(i, x, n) for(int i = x; i >= n; i--)
#define elif else if
#define sp(x) fixed << setprecision(x)
#define pb push_back
#define eb emplace_back
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
using ll = long long;
using ld = long double;
using pii = pair<int, int>;
using pil = pair<int, ll>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
const ll MOD = 1e9+7;
//const ll MOD = 998244353;
const int inf = (1<<30)-1;
const ll INF = (1LL<<60)-1;
const ld EPS = 1e-10;
template<typename T> bool chmax(T &x, const T &y) {return (x < y)? (x = y, true) : false;};
template<typename T> bool chmin(T &x, const T &y) {return (x > y)? (x = y, true) : false;};

template<typename Monoid> struct Lazy_Segment_Tree{
    Monoid ope(Monoid a, Monoid b){
        return a+b;
    }
    Monoid unit;
    int n;
    vector<Monoid> seg, lazy, sqsum;

    Lazy_Segment_Tree(int N, Monoid x){
        unit = x;
        n = 1;
        while(n < N) n <<= 1;
        seg.assign(2*n, unit);
        lazy.assign(2*n, unit);
        sqsum.assign(2*n, unit);
    }
    
    void eval(int i, int l, int r){
        if(lazy[i] != unit){
            sqsum[i] = ope(sqsum[i], 2*seg[i]*lazy[i] + (r-l)*lazy[i]*lazy[i]);
            seg[i] = ope(seg[i], (r-l)*lazy[i]);
            if(r-l > 1){
                lazy[2*i] = ope(lazy[2*i] ,lazy[i]);
                lazy[2*i+1] = ope(lazy[2*i+1], lazy[i]);
            }
            lazy[i] = unit;
        }
    }
    
    void add(int a, int b, Monoid x, int i = 1, int l = 0, int r = -1){
        if(r < 0) r = n;
        eval(i, l, r);
        if(a >= r || b <= l) return;
        elif(a <= l && r <= b){
            lazy[i] = ope(lazy[i], x);
            eval(i, l, r);
        }
        else{
            add(a, b, x, 2*i, l, (l+r)/2);
            add(a, b, x, 2*i+1, (l+r)/2, r);
            seg[i] = ope(seg[2*i], seg[2*i+1]);
            sqsum[i] = ope(sqsum[2*i], sqsum[2*i+1]);
        }
    }
    
    Monoid query(int a, int b, int i = 1, int l = 0, int r = -1){
        if(r < 0) r = n;
        if(a >= r || b <= l) return unit;
        eval(i, l, r);
        if(a <= l && r <= b) return sqsum[i];
        Monoid vl = query(a, b, 2*i, l, (l+r)/2);
        Monoid vr = query(a, b, 2*i+1, (l+r)/2, r);
        return ope(vl, vr);
    }
    
    void update(int i = 1, int l = 0, int r = -1){
        if(r < 0) r = n;
        seg[i] = ope(seg[i], lazy[i]);
        if(r-l > 1){
            lazy[2*i] = ope(lazy[2*i], lazy[i]/2);
            lazy[2*i+1] = ope(lazy[2*i+1], lazy[i]/2);
            update(2*i, l, (l+r)/2);
            update(2*i+1, (l+r)/2, r);
        }
        lazy[i] = unit;
    }
    
    Monoid at(int i){
        return seg[n+i];
    }
    
    void clear(){
        fill(seg.begin(), seg.end(), unit);
        fill(lazy.begin(), lazy.end(), unit);
    }
};

int main(){
    int N;
    cin >> N;
    Lazy_Segment_Tree<ll> seg(N, 0);
    ll A[N];
    rep(i, N){
        cin >> A[i];
        seg.add(i, i+1, A[i]);
    }
    int Q;
    cin >> Q;
    rep(i, Q){
        int a, l, r, x;
        cin >> a >> l >> r;
        if(a == 1){
            cin >> x;
            seg.add(l-1, r, x);
        }
        else{
            cout << seg.query(l-1, r) << endl;
        }
    }
}
0