結果

問題 No.776 A Simple RMQ Problem
ユーザー IKyoproIKyopro
提出日時 2019-12-03 18:06:31
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 343 ms / 3,000 ms
コード長 2,963 bytes
コンパイル時間 1,034 ms
コンパイル使用メモリ 96,036 KB
実行使用メモリ 11,520 KB
最終ジャッジ日時 2024-05-05 13:55:03
合計ジャッジ時間 10,911 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 34 ms
5,376 KB
testcase_03 AC 123 ms
11,392 KB
testcase_04 AC 149 ms
5,376 KB
testcase_05 AC 256 ms
11,392 KB
testcase_06 AC 78 ms
7,424 KB
testcase_07 AC 198 ms
11,520 KB
testcase_08 AC 204 ms
7,424 KB
testcase_09 AC 76 ms
11,392 KB
testcase_10 AC 109 ms
7,424 KB
testcase_11 AC 203 ms
11,520 KB
testcase_12 AC 259 ms
11,520 KB
testcase_13 AC 260 ms
11,520 KB
testcase_14 AC 260 ms
11,520 KB
testcase_15 AC 262 ms
11,520 KB
testcase_16 AC 270 ms
11,520 KB
testcase_17 AC 343 ms
11,392 KB
testcase_18 AC 189 ms
11,392 KB
testcase_19 AC 303 ms
11,520 KB
testcase_20 AC 299 ms
11,392 KB
testcase_21 AC 322 ms
11,520 KB
testcase_22 AC 327 ms
11,392 KB
testcase_23 AC 334 ms
11,392 KB
testcase_24 AC 304 ms
11,392 KB
testcase_25 AC 99 ms
5,376 KB
testcase_26 AC 201 ms
11,520 KB
testcase_27 AC 177 ms
11,520 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <climits>
#include <functional>
#include <algorithm>
using namespace std;
using ll = long long;
template<class T> using vec = vector<T>;
template<class T> using vvec = vector<vec<T>>;

template<typename Monoid>
class SegmentTree{
private:
    using F = function<Monoid(Monoid&,Monoid&)>;
    int sz;
    vector<Monoid> seg;
    const F op;//演算
    const Monoid e;//単位元
public:
    SegmentTree(int n,const F op,const Monoid &e):op(op),e(e){
        sz = 1;
        while(sz<n) sz <<= 1;
        seg.assign(2*sz,e);
    }
    //代入
    void set(int k, const Monoid &x){
        seg[k+sz] = x;
    }
    //前計算(?)
    void build(){
        for(int i=sz-1;i>0;i--){
            seg[i] = op(seg[2*i],seg[2*i+1]);
        }
    }
    void update(int k,const Monoid &x){
        k += sz;
        seg[k] = x;
        while(k>>=1){
            seg[k] = op(seg[2*k],seg[2*k+1]);
        }
    }
    Monoid query(int l,int r){
        Monoid L = e,R = e;
        for(l+=sz,r+=sz;l<r;l>>=1,r>>=1){
            if(l&1) L = op(L,seg[l++]);
            if(r&1) R = op(seg[--r],R);
        }
        return op(L,R);
    }
    Monoid operator[](const int &k)const{
        return seg[k+sz];
    }
};

struct state{
    ll sum,lsum,rsum,ma;
};

constexpr ll inf = 1e18;

state op(state& l,state& r){
    ll sum,lsum,rsum,ma;
    sum = l.sum+r.sum;
    lsum = max(l.lsum,l.sum+r.lsum);
    rsum = max(r.rsum,r.sum+l.rsum);
    ma = max(max(l.ma,r.ma),l.rsum+r.lsum);
    return (state){sum,lsum,rsum,ma};
}

int main(){
    int N,Q;
    cin >> N >> Q;
    SegmentTree<state> seg(N,op,(state){0,-inf,-inf,-inf});
    for(int i=0;i<N;i++){
        ll a;
        cin >> a;
        seg.set(i,(state){a,a,a,a});
    }
    seg.build();
    for(int i=0;i<Q;i++){
        string S;
        cin >> S;
        if(S=="set"){
            int id; ll val;
            cin >> id >> val;
            id--;
            seg.update(id,(state){val,val,val,val});
        }else{
            vec<int> L(2),R(2);
            cin >> L[0] >> L[1] >> R[0] >> R[1];
            L[0]--; L[1]--; R[0]--; R[1]--;
            if(L[1]<R[0]){
                cout << seg.query(L[0],L[1]+1).rsum+seg.query(L[1]+1,R[0]).sum+seg.query(R[0],R[1]+1).lsum << endl;
            }else if(L[0]<=R[0] && R[1]<=L[1]){
                cout << max(seg.query(L[0],R[0]).rsum+seg.query(R[0],R[1]+1).lsum,seg.query(R[0],R[1]+1).ma) << endl;
            }else if(R[0]<=L[0] && L[1]<=R[1]){
                cout << max(seg.query(L[0],L[1]+1).ma,seg.query(L[0],L[1]+1).rsum+seg.query(L[1]+1,R[1]+1).lsum) << endl;
            }else if(R[0]<=L[0]){
                cout << seg.query(L[0],R[1]+1).ma << endl;
            }else{
                cout << max({seg.query(L[0],R[0]).rsum+seg.query(R[0],R[1]+1).lsum,
                seg.query(L[0],L[1]+1).rsum+seg.query(L[1]+1,R[1]+1).lsum,
                seg.query(R[0],L[1]+1).ma}) << endl;
            }
        }
    }
}
0