結果

問題 No.1099 Range Square Sum
ユーザー Navier_Boltzmann
提出日時 2023-12-31 13:08:15
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,118 ms / 2,000 ms
コード長 1,219 bytes
コンパイル時間 4,988 ms
コンパイル使用メモリ 319,172 KB
実行使用メモリ 44,972 KB
最終ジャッジ日時 2024-09-27 17:05:12
合計ジャッジ時間 17,968 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>


using namespace std;
using namespace atcoder;
#define rep(i,m,n) for (int i = (int)(m); i < (int)(n); i++)
#define ll long long

vector<ll> e(){
    
    vector<ll> _e(3);
    return _e;
};

vector<ll> op(vector<ll> x, vector<ll> y){

    vector<ll> z(3);
    rep(i,0,3){
        z[i] = x[i] + y[i];
    };
    return z;
}

vector<ll> mapping(ll f, vector<ll> x){

    vector<ll> z(3);
    z[0] = x[0];
    z[1] = x[1] + f*x[0];
    z[2] = x[2] + 2*x[1]*f + x[0]*(f*f);

    return z; 
};
ll composition(ll f, ll g){
    return f+g;
};
ll id_(){
    return 0LL;
};



int main(){

    int N,Q;
    cin >> N ;
    vector<vector<ll>> A(N);
    vector<ll> a(3);
    ll tmp;
    rep(i,0,N){
        cin >> tmp;
        a[0] = 1;
        a[1] = tmp;
        a[2] = tmp*tmp;
        A[i] = a;
    };
    cin >> Q;
    lazy_segtree<vector<ll>,op,e,ll,mapping,composition, id_> T(A);
    int q,l,r;
    ll x;
    rep(_,0,Q){
        cin >> q;
        if (q==1){
            cin >> l >> r >> x;
            l -= 1;
            T.apply(l,r,x);
        }
        else{
            cin >> l >> r ;
            l -= 1;
            cout << T.prod(l,r)[2] << endl;
        };
    }

}
0