結果

問題 No.1099 Range Square Sum
ユーザー kotatsugame
提出日時 2020-06-26 21:54:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 549 ms / 2,000 ms
コード長 2,542 bytes
コンパイル時間 895 ms
コンパイル使用メモリ 87,944 KB
実行使用メモリ 22,912 KB
最終ジャッジ日時 2024-07-04 20:57:05
合計ジャッジ時間 7,160 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:80:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   80 | main()
      | ^~~~

ソースコード

diff #
プレゼンテーションモードにする

#include<iostream>
using namespace std;
#include<vector>
#include<functional>
template<typename T>
struct lazysegtree{
    using F=function<T(T,T)>;
    using G=function<T(T,T,int,int)>;
    const F calcfn,lazycalcfn;
    const G updatefn;
    int n;
    T defvalue,lazydefvalue;
    vector<T>dat,lazy;
    vector<bool>lazyflag;
    lazysegtree(int n_=0,T defvalue_=0,
        const F calcfn_=[](T a,T b){return a+b;},
        const F lazycalcfn_=[](T a,T b){return a+b;},
        const G updatefn_=[](T a,T b,int l,int r){return a+b*(r-l);},
        T lazydefvalue_=0
    ):defvalue(defvalue_),lazydefvalue(lazydefvalue_),
        calcfn(calcfn_),lazycalcfn(lazycalcfn_),updatefn(updatefn_)
    {
        n=1;
        while(n<n_)n<<=1;
        dat.assign(2*n-1,defvalue);
        lazy.assign(2*n-1,lazydefvalue);
        lazyflag.assign(2*n-1,false);
    }
    void copy(const vector<T>&v)
    {
        for(int i=0;i<v.size();i++)dat[i+n-1]=v[i];
        for(int i=n-2;i>=0;i--)dat[i]=calcfn(dat[2*i+1],dat[2*i+2]);
    }
    void eval(int i,int l,int r)
    {
        if(lazyflag[i])
        {
            dat[i]=updatefn(dat[i],lazy[i],l,r);
            if(r-l>1)
            {
                lazy[2*i+1]=lazycalcfn(lazy[2*i+1],lazy[i]);
                lazy[2*i+2]=lazycalcfn(lazy[2*i+2],lazy[i]);
                lazyflag[2*i+1]=lazyflag[2*i+2]=true;
            }
            lazy[i]=lazydefvalue;
            lazyflag[i]=false;
        }
    }
    void update(int a,int b,T x,int k=0,int l=0,int r=-1)//[a,b)
    {
        if(r<0)r=n;
        eval(k,l,r);
        if(b<=l||r<=a)return;
        else if(a<=l&&r<=b)
        {
            lazy[k]=lazycalcfn(lazy[k],x);
            lazyflag[k]=true;
            eval(k,l,r);
        }
        else
        {
            update(a,b,x,2*k+1,l,(l+r)/2);
            update(a,b,x,2*k+2,(l+r)/2,r);
            dat[k]=calcfn(dat[2*k+1],dat[2*k+2]);
        }
    }
    T query(int a,int b,int k=0,int l=0,int r=-1)//[a,b)
    {
        if(r<0)r=n;
        eval(k,l,r);
        if(b<=l||r<=a)return defvalue;
        else if(a<=l&&r<=b)return dat[k];
        else return calcfn(
            query(a,b,2*k+1,l,(l+r)/2),
            query(a,b,2*k+2,(l+r)/2,r)
        );
    }
};
int N;
main()
{
    cin>>N;
    vector<pair<long,long> >A(N);
    for(int i=0;i<N;i++)
    {
        cin>>A[i].first;
        A[i].second=A[i].first*A[i].first;
    }
    lazysegtree<pair<long,long> >P(N,make_pair(0L,0L),
    [](pair<long,long>a,pair<long,long>b){return make_pair(a.first+b.first,a.second+b.second);},
    [](pair<long,long>a,pair<long,long>b){return make_pair(a.first+b.first,0L);},
    [](pair<long,long>a,pair<long,long>b,int l,int r){
        return make_pair(a.first+b.first*(r-l),a.second+2*a.first*b.first+b.first*b.first*(r-l));
    },make_pair(0L,0L));
    P.copy(A);
    int Q;cin>>Q;
    for(;Q--;)
    {
        int q,l,r;cin>>q>>l>>r;
        if(q==1)
        {
            int x;cin>>x;
            P.update(l-1,r,make_pair(x,0));
        }
        else
        {
            cout<<P.query(l-1,r).second<<endl;
        }
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0