結果

問題 No.876 Range Compress Query
ユーザー SugarDragon5SugarDragon5
提出日時 2019-09-06 22:52:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 391 ms / 2,000 ms
コード長 4,994 bytes
コンパイル時間 2,308 ms
コンパイル使用メモリ 177,448 KB
実行使用メモリ 11,596 KB
最終ジャッジ日時 2023-09-07 01:49:15
合計ジャッジ時間 6,081 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 3 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 3 ms
4,384 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,384 KB
testcase_06 AC 3 ms
4,380 KB
testcase_07 AC 3 ms
4,376 KB
testcase_08 AC 3 ms
4,380 KB
testcase_09 AC 3 ms
4,384 KB
testcase_10 AC 3 ms
4,380 KB
testcase_11 AC 380 ms
11,596 KB
testcase_12 AC 314 ms
11,404 KB
testcase_13 AC 312 ms
11,344 KB
testcase_14 AC 371 ms
11,316 KB
testcase_15 AC 260 ms
11,044 KB
testcase_16 AC 365 ms
11,320 KB
testcase_17 AC 361 ms
11,468 KB
testcase_18 AC 391 ms
11,016 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) for(int i=0;i<(n);i++)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
template<typename A,typename B>istream&operator>>(istream&is,pair<A,B> &p){is>>p.fi>>p.se;return is;}
template<typename A,typename B>ostream&operator<<(ostream&os,const pair<A,B> &p){os<<"("<<p.fi<<","<<p.se<<")";return os;}
template<typename T>istream&operator>>(istream&is,vector<T> &t){REP(i,t.size())is>>t[i];return is;}
template<typename T>ostream&operator<<(ostream&os,const vector<T>&t){os<<"{";REP(i,t.size()){if(i)os<<",";os<<t[i];}cout<<"}";return os;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;

struct data{
    ll l,r;
    int x;
    data(ll l,ll r,ll x):l(l),r(r),x(x){}
    bool operator==(const data& o){return l==o.l&&r==o.r&&x==o.x;}
};
ostream&operator<<(ostream&os,const data&d){cerr<<"{{"<<d.l<<","<<d.r<<"},"<<d.x<<"}";return os;}

template<typename Monoid,typename OperatorMonoid=Monoid>
struct LazySegTree{
    //0-indexed 閉区間
    using F=function<Monoid(Monoid,Monoid)>;
    using G=function<Monoid(Monoid,OperatorMonoid)>;
    using H=function<OperatorMonoid(OperatorMonoid,OperatorMonoid)>;
    using P=function<OperatorMonoid(OperatorMonoid,int)>;

    int size;
    vector<Monoid> data;
    vector<OperatorMonoid> lazy;
    const F f;  //要素同士の演算
    const G g;  //要素と作用素のマージ
    const H h;  //作用素同士のマージ
    const P p;  //作用素を下に降ろす
    const Monoid M1;    //単位元
    const OperatorMonoid OM0;   //作用素の単位元

    LazySegTree(int n,const F f,const G g,const H h,const P p,const Monoid &M1,const OperatorMonoid OM0)
        :f(f),g(g),h(h),p(p),M1(M1),OM0(OM0){
        size=1;
        while(size<n)size<<=1;
        data.assign(size*2-1,M1);
        lazy.assign(size*2-1,OM0);
    }

    void set(int k,const Monoid &x){
        //kをxに初期化
        //buildを忘れずに行うこと
        data[k+size-1]=x;
    }
    void build(){
        //構築 O(N)
        for(int k=size-2;k>=0;k--){
            data[k]=f(data[2*k+1],data[2*k+2]);
        }
    }
    void propagate(int k,int len){
        //長さlenのkから下へ伝播
        if(lazy[k]!=OM0){
            if(k<size-1){
                lazy[2*k+1]=h(lazy[2*k+1],lazy[k]);
                lazy[2*k+2]=h(lazy[2*k+2],lazy[k]);
            }
            data[k]=g(data[k],p(lazy[k],len));
            lazy[k]=OM0;
        }
    }

    Monoid update(int a,int b,const OperatorMonoid &x,int k,int l,int r){
        propagate(k,r-l+1);
        if(r<a||b<l){ //範囲外
            return data[k];
        }else if(a<=l&&b>=r){   //内部
            lazy[k]=h(lazy[k],x);
            propagate(k,r-l+1);
            return data[k];
        }else{
            return data[k]=f(update(a,b,x,2*k+1,l,(l+r)/2),
                    update(a,b,x,2*k+2,(l+r)/2+1,r));
        }
    }
    Monoid update(int a,int b,const OperatorMonoid &x){
        //[a,b]をxに変更
        return update(a,b,x,0,0,size-1);
    }

    Monoid query(int a,int b,int k,int l,int r){
        propagate(k,r-l+1);
        if(r<a||b<l){
            return M1;
        }else if(a<=l&&b>=r){
            return data[k];
        }else{
            return f(query(a,b,2*k+1,l,(l+r)/2),
                    query(a,b,2*k+2,(l+r)/2+1,r));
        }
    }
    Monoid query(int a,int b){
        return query(a,b,0,0,size-1);
    }

    Monoid operator[](const int &k){
        return query(k,k);
    }
    void debug(){
        for(int i=0;i<size;i++){
            if(i)cerr<<" ";cerr<<(*this)[i];
        }cerr<<endl;
    }
};

int main(){
    int n,q;
    cin>>n>>q;
    auto f=[](data a,data b){
        if(a==data(-1ll,-1ll,0ll))return b;
        if(b==data(-1ll,-1ll,0ll))return a;
        data t{a.l,b.r,a.x+b.x};
        if(a.r==b.l)t.x--;
        return t;
    };
    auto g=[](data a,ll x){a.l+=x;a.r+=x;return a;};
    auto h=[](ll x,ll y){return x+y;};
    auto p=[](ll a,ll b){return a;};
    LazySegTree<data,ll> a(n,f,g,h,p,{-1,-1,0},0ll);
    REP(i,n){
        int t;
        cin>>t;
        a.set(i,{t,t,1});
    }
    a.build();
    REP(i,q){
        int f;
        cin>>f;
        if(f==1){
            int l,r;ll x;
            cin>>l>>r>>x;
            l--;r--;
            a.update(l,r,x);
        }else{
            int l,r;
            cin>>l>>r;
            l--;r--;
            cout<<a.query(l,r).x<<endl;
        }
    }
    return 0;
}
0