#671657 (C++17) No.776 A Simple RMQ Problem

提出ソース
結果

問題	No.776 A Simple RMQ Problem
ユーザー	tsuyu93
提出日時	2021-07-01 17:37:30
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	110 ms / 3,000 ms
コード長	18,277 bytes
コンパイル時間	2,927 ms
コンパイル使用メモリ	228,388 KB
実行使用メモリ	14,556 KB
最終ジャッジ日時	2024-06-28 06:28:30
合計ジャッジ時間	9,443 ms
ジャッジサーバーID （参考情報）	judge3 / judge5
このコードへのチャレンジ
（要ログイン）
テストケース

テストケース表示
入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	15 ms 6,940 KB
testcase_03	AC	51 ms 13,664 KB
testcase_04	AC	51 ms 6,940 KB
testcase_05	AC	89 ms 13,884 KB
testcase_06	AC	29 ms 8,312 KB
testcase_07	AC	63 ms 14,120 KB
testcase_08	AC	65 ms 8,428 KB
testcase_09	AC	37 ms 14,092 KB
testcase_10	AC	41 ms 8,552 KB
testcase_11	AC	75 ms 14,356 KB
testcase_12	AC	94 ms 14,408 KB
testcase_13	AC	93 ms 14,372 KB
testcase_14	AC	92 ms 14,368 KB
testcase_15	AC	93 ms 14,440 KB
testcase_16	AC	94 ms 14,504 KB
testcase_17	AC	110 ms 14,492 KB
testcase_18	AC	80 ms 14,540 KB
testcase_19	AC	100 ms 14,388 KB
testcase_20	AC	99 ms 14,548 KB
testcase_21	AC	100 ms 14,488 KB
testcase_22	AC	100 ms 14,556 KB
testcase_23	AC	102 ms 14,388 KB
testcase_24	AC	99 ms 14,388 KB
testcase_25	AC	27 ms 6,940 KB
testcase_26	AC	81 ms 14,388 KB
testcase_27	AC	75 ms 14,428 KB
権限があれば一括ダウンロードができます
ソースコード

raw source code
/**
 * author:  tsuyu93
 * started: 2021-07-01  13:32:05
 * 継続は力なり、雨だれ石を穿つ
 * make the impossible possible
**/

#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>

// #include <boost/multiprecision/cpp_dec_float.hpp>
// #include <boost/multiprecision/cpp_int.hpp>
// namespace mpb = boost::multiprecision;
// using bint = mpb::cpp_int;
// // 仮数部が1024ビットの浮動小数点数型(TLEしたら小さくする)
// using Real = mpb::number<mpb::cpp_dec_float<1024>>;

#ifdef __HOGE__
#pragma region macro
#endif
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
using uint = unsigned;
using pll = pair<ll, ll>;
using pli = pair<ll, int>;
using pii = pair<int, int>;
using pld = pair<ll, ld>;
using ppiii =  pair<pii, int>;
using ppiill = pair<pii, ll>;
using ppllll = pair<pll, ll>;
using pplii = pair<pli, int>;
using mii = map<int, int>;
using dll = deque<ll>;
using qll = queue<ll>;
using pqll = priority_queue<ll>;
using pqrll = priority_queue<ll, vector<ll>, greater<ll>>;
using pqrpll = priority_queue<pll, vector<pll>, greater<pll>>;
using vint = vector<int>;
using vbool = vector<bool>;
using vstr = vector<string>;
using vll = vector<ll>;
using vld = vector<ld>;
using vpll = vector<pll>;
using vvll = vector<vector<ll>>;
using vvint = vector<vector<int>>;
using vvld = vector<vector<ld>>;
using vvbool = vector<vbool>;
using vvstr = vector<vstr>;
using vvpll = vector<vector<pll>>;
#define REP(i,n) for(ll i=0;i<ll(n);i++)
#define REPD(i,n) for(ll i=n-1;i>=0;i--)
#define FOR(i,a,b) for(ll i=a;i<=ll(b);i++)
#define FORD(i,a,b) for(ll i=a;i>=ll(b);i--)
#define overload4(_1,_2,_3,_4,name,...) name
#define overload3(_1,_2,_3,name,...) name
#define rep1(n) for(ll i=0;i<n;++i)
#define rep2(i,n) for(ll i=0;i<n;++i)
#define rep3(i,a,b) for(ll i=a;i<b;++i)
#define rep4(i,a,b,c) for(ll i=a;i<b;i+=c)
#define rep(...) overload4(__VA_ARGS__,rep4,rep3,rep2,rep1)(__VA_ARGS__)
#define rrep1(n) for(ll i=n;i--;)
#define rrep2(i,n) for(ll i=n;i--;)
#define rrep3(i,a,b) for(ll i=b;i-->(a);)
#define rrep4(i,a,b,c) for(ll i=(a)+((b)-(a)-1)/(c)*(c);i>=(a);i-=c)
#define rrep(...) overload4(__VA_ARGS__,rrep4,rrep3,rrep2,rrep1)(__VA_ARGS__)
#define each1(i,a) for(auto&&i:a)
#define each2(x,y,a) for(auto&&[x,y]:a)
#define each3(x,y,z,a) for(auto&&[x,y,z]:a)
#define each(...) overload4(__VA_ARGS__,each3,each2,each1)(__VA_ARGS__)
#define all1(i) begin(i),end(i)
#define all2(i,a) begin(i),begin(i)+a
#define all3(i,a,b) begin(i)+a,begin(i)+b
#define all(...) overload3(__VA_ARGS__,all3,all2,all1)(__VA_ARGS__)
#define rall1(i) (i).rbegin(),(i).rend()
#define rall2(i,k) (i).rbegin(),(i).rbegin()+k
#define rall3(i,a,b) (i).rbegin()+a,(i).rbegin()+b
#define rall(...) overload3(__VA_ARGS__,rall3,rall2,rall1)(__VA_ARGS__)
#define SUM(...) accumulate(all(__VA_ARGS__),0LL)
#define ALL(x) x.begin(),x.end() 
#define rALL(x) x.rbegin(),x.rend() 
#define SIZE(x) ll(x.size()) 
#define Sort(a) sort(all(a))
#define INT(...) int __VA_ARGS__;in(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;in(__VA_ARGS__)
#define ULL(...) ull __VA_ARGS__;in(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;in(__VA_ARGS__)
#define CHR(...) char __VA_ARGS__;in(__VA_ARGS__)
#define DBL(...) double __VA_ARGS__;in(__VA_ARGS__)
#define LD(...) ld __VA_ARGS__;in(__VA_ARGS__)int scan(){ return getchar(); }
#define fs first
#define sc second
#define endl '\n'
#define elif else if
#define ADD_OVERFLOW(a, b) __builtin_add_overflow_p (a, b, (decltype((a)+(b))) 0)
#define SUB_OVERFLOW(a, b) __builtin_sub_overflow_p (a, b, (decltype((a)+(b))) 0)
#define MUL_OVERFLOW(a, b) __builtin_mul_overflow_p (a, b, (decltype((a)+(b))) 0)
void scan(int& a){ scanf("%d", &a); }
void scan(unsigned& a){ scanf("%u", &a); }
void scan(long& a){ scanf("%ld", &a); }
void scan(long long& a){ scanf("%lld", &a); }
void scan(unsigned long long& a){ scanf("%llu", &a); }
void scan(char& a){ do{ a = getchar(); }while(a == ' ' || a == '\n'); }
void scan(float& a){ scanf("%f", &a); }
void scan(double& a){ scanf("%lf", &a); }
void scan(long double& a){ scanf("%Lf", &a); }
void scan(vector<bool>& a){ for(unsigned i = 0; i < a.size(); i++){ int b; scan(b); a[i] = b; } }
void scan(char a[]){ scanf("%s", a); }
void scan(string& a){ cin >> a; }
template<class T> void scan(vector<T>&);
template<class T, size_t size> void scan(array<T, size>&);
template<class T, class L> void scan(pair<T, L>&);
template<class T, size_t size> void scan(T(&)[size]);
template<class T> void scan(vector<T>& a){ for(auto&& i : a) scan(i); }
template<class T> void scan(deque<T>& a){ for(auto&& i : a) scan(i); }
template<class T, size_t size> void scan(array<T, size>& a){ for(auto&& i : a) scan(i); }
template<class T, class L> void scan(pair<T, L>& p){ scan(p.first); scan(p.second); }
template<class T, size_t size> void scan(T (&a)[size]){ for(auto&& i : a) scan(i); }
template<class T> void scan(T& a){ cin >> a; }
void in(){}
template <class Head, class... Tail> void in(Head& head, Tail&... tail){ scan(head); in(tail...); }
void print(){ putchar(' '); }
void print(bool a){ printf("%d", a); }
void print(int a){ printf("%d", a); }
void print(unsigned a){ printf("%u", a); }
void print(long a){ printf("%ld", a); }
void print(long long a){ printf("%lld", a); }
void print(unsigned long long a){ printf("%llu", a); }
void print(char a){ printf("%c", a); }
void print(char a[]){ printf("%s", a); }
void print(const char a[]){ printf("%s", a); }
void print(float a){ printf("%.15f", a); }
void print(double a){ printf("%.15f", a); }
void print(long double a){ printf("%.15Lf", a); }
void print(const string& a){ for(auto&& i : a) print(i); }
template<class T> void print(const complex<T>& a){ if(a.real() >= 0) print('+'); print(a.real()); if(a.imag() >= 0) print('+'); print(a.imag()); print('i'); }
template<class T> void print(const vector<T>&);
template<class T, size_t size> void print(const array<T, size>&);
template<class T, class L> void print(const pair<T, L>& p);
template<class T, size_t size> void print(const T (&)[size]);
template<class T> void print(const vector<T>& a){ if(a.empty()) return; print(a[0]); for(auto i = a.begin(); ++i != a.end(); ){ putchar(' '); print(*i); } }
template<class T> void print(const deque<T>& a){ if(a.empty()) return; print(a[0]); for(auto i = a.begin(); ++i != a.end(); ){ putchar(' '); print(*i); } }
template<class T, size_t size> void print(const array<T, size>& a){ print(a[0]); for(auto i = a.begin(); ++i != a.end(); ){ putchar(' '); print(*i); } }
template<class T, class L> void print(const pair<T, L>& p){ print(p.first); putchar(' '); print(p.second); }
template<class T, size_t size> void print(const T (&a)[size]){ print(a[0]); for(auto i = a; ++i != end(a); ){ putchar(' '); print(*i); } }
template<class T> void print(const T& a){ cout << a; }
int out(){ putchar('\n'); return 0; }
template<class T> int out(const T& t){ print(t); putchar('\n'); return 0; }
template<class Head, class... Tail> int out(const Head& head, const Tail&... tail){ print(head); putchar(' '); out(tail...); return 0; }
template <class T> vector<T> make_vec(size_t a, T b) {return vector<T>(a, b);}
template <class T, class... Ts> auto make_vec(size_t a, Ts... ts) {return vector<decltype(make_vec<T>(ts...))>(a, make_vec<T>(ts...));}
#ifdef DEBUG
inline ll __lg(ull x){ return 63 - __builtin_clzll(x); }
#define debug(...) { print(#__VA_ARGS__); print(":"); out(__VA_ARGS__); }
#else
#define debug(...) void(0)
#endif
#define INF32 2147483647 //2.147483647x10^{9}:32bit整数のinf
#define INF64 9223372036854775807 //9.223372036854775807x10^{18}:64bit整数のinf
const ll MOD = 1000000007;
// const ll MOD = 998244353;
const int inf = 1e9;
const ll INF = (1LL<<61);
const ll MAXR = 100000; //10^5:配列の最大のrange
const ld PI = acos(-1);
inline void Yes(bool b = true) { cout << (b ? "Yes" : "No") << '\n'; }
inline void YES(bool b = true) { cout << (b ? "YES" : "NO") << '\n'; }
inline void OKNG(bool b = true) { cout << (b ? "OK" : "NG") << '\n'; }
inline void GAme(bool b = true) { cout << (b ? "First" : "Second") << '\n'; }
inline void gAme(bool b = true) { cout << (b ? "first" : "second") << '\n'; }
inline void Takahashi_Aoki(bool b = true) { cout << (b ? "Takahashi" : "Aoki") << '\n'; }
inline void takahashi_aoki(bool b = true) { cout << (b ? "takahashi" : "aoki") << '\n'; }
inline void possible(bool i = true){ cout << (i?"possible":"impossible") << '\n'; }
inline void Possible(bool i = true){ cout << (i?"Possible":"Impossible") << '\n'; }
inline void POSSIBLE(bool i = true){ cout << (i?"POSSIBLE":"IMPOSSIBLE") << '\n'; }
template<typename A, typename B> inline bool chmin(A& a, B b) {if (a > b) {a = b;return true;}return false;}
template<typename A, typename B> inline bool chmax(A& a, B b) {if (a < b) {a = b;return true;}return false;}
template<class T> auto min(const T& a){ return *min_element(a.begin(),a.end()); }
template<class T> auto max(const T& a){ return *max_element(a.begin(),a.end()); }
template<typename T> T power(T a, T b) {T res=1;while (b>0) {if (b&1){res*=a;}b/=2;a*=a;}return res;}
long long powmod(long long a, long long b) {long long res=1;while (b>0) {if (b&1){res*=a;res%=MOD;}b/=2;a*=a;a%=MOD;}return res;}
template<typename T> T powmod(T a, T b,T m) {T res=1;while (b>0) {if (b&1){res*=a;res%=m;}b/=2;a*=a;a%=m;}return res%m;}
template<typename T> T nlcm(const vector<T> &a) {T res;res = a[0];for (ll i = 1; i < (ll)a.size(); i++) {res = lcm(res, a[i]);}return res;}
template<typename T> T ngcd(const vector<T> &a){T res;res = a[0];for(ll i = 1; i < (ll)a.size() && res != 1; i++) {res = gcd(a[i], res);}return res;}
template<typename T> T sqr(T x){T r=sqrt(x)-1;while((r+1)*(r+1)<=x)r++;return r;}
template<typename T> T bit_length(T a){T k = 0;while(a){k++;a/=2;}return k;}
template<typename T> T isqrt(T n){T a = 0, r = 0;for(T s = bit_length(n)-(bit_length(n)%2?1:2); s >= 0;s-=2){T t = ((n >> s) & 3);r = ((r << 2) | t);T c = ((a << 2) | 1);int b = (r >= c);if(b){r -= c;}a = ((a << 1) | b);}return a;}
int popcount(long long a){return __builtin_popcountll(a);}
int popcount(int a){return __builtin_popcount(a);}
template<typename T> T ceil(T a, T b) {if(b < 0) a = -a, b = -b;return (a >= 0 ? (a + b - 1) / b : a / b);}
template<typename T> T floor(T a, T b) {if(b < 0) a = -a, b = -b;return (a >= 0 ? a / b : (a - b + 1) / b);}
template<typename T> T sum_of_digit(T a) {T res = 0;while(a){res += a%10; a /= 10;}return res;}
template<typename T ,typename A> void add_edge(vector<vector<T>> &G, A a, A b,bool directed = false){G[a].emplace_back(b);if(!directed)G[b].emplace_back(a);}
template<typename T ,typename A> void add_edge(vector<vector<pair<T,T>>> &G, A a, A b, A p,bool directed = false){G[a].emplace_back(p,b);if(!directed)G[b].emplace_back(p,a);}
#ifdef __HOGE__
#pragma endregion
#endif

// #include<atcoder/all>
// using namespace atcoder;
// using mint = modint1000000007;
// using mint = modint998244353;

template <typename typeT = std::nullptr_t, typename typeE = std::nullptr_t, int seg_type = 1>
class segment_tree{
    static_assert(1 <= seg_type && seg_type <= 3);

protected:
    virtual typeT e_t(){return typeT();};
    virtual typeE e_e(){return typeE();};
    virtual typeT func_tt(typeT a, typeT b){return e_t();};
    virtual typeT func_te(typeT t, typeE e){return e_t();};
    virtual typeE func_ee(typeE a, typeE b){return e_e();};

    unsigned int num, lognum = 0, sz;
    std::vector<typeT> seg_t;
    std::vector<typeE> seg_e;

    void preset(unsigned int n, std::conditional_t<seg_type == 2, typeE, typeT> def){preset(std::vector(n, def));}

    void preset(std::vector<std::conditional_t<seg_type == 2, typeE, typeT>> vec){
        num = vec.size();
        while((1U << lognum) < num) lognum++;
        sz = 1U << lognum;

        if constexpr(seg_type == 1){
            seg_t = std::vector<typeT>(sz * 2, e_t());
            std::copy(vec.begin(), vec.end(), seg_t.begin() + sz);
            for(unsigned int i = sz - 1; i > 0; i--) update_t(i);
        }
        else if constexpr(seg_type == 2){
            seg_e = std::vector<typeE>(sz * 2, e_e());
            std::copy(vec.begin(), vec.end(), seg_e.begin() + sz);
        }
        else if constexpr(seg_type == 3){
            seg_t = std::vector<typeT>(sz * 2, e_t());
            seg_e = std::vector<typeE>(sz, e_e());
            std::copy(vec.begin(), vec.end(), seg_t.begin() + sz);
            for(unsigned int i = sz - 1; i > 0; i--) update_t(i);
        }
    }

    void update_t(unsigned int ind){
        if constexpr(seg_type != 2){
            seg_t[ind] = func_tt(seg_t[ind * 2], seg_t[ind * 2 + 1]);
        }
    }
    
    void apply_e(unsigned int ind, typeE query){
        if constexpr(seg_type == 1){
            seg_t[ind] = func_te(seg_t[ind], query);
        }
        else if constexpr(seg_type == 2){
            seg_e[ind] = func_ee(seg_e[ind], query);
        }
        else if constexpr(seg_type == 3){
            seg_t[ind] = func_te(seg_t[ind], query);
            if(ind < sz) seg_e[ind] = func_ee(seg_e[ind], query);
        }
    }

    void push_e(unsigned int ind){
        if constexpr(seg_type != 1){
            apply_e(ind * 2, seg_e[ind]);
            apply_e(ind * 2 + 1, seg_e[ind]);
            seg_e[ind] = e_e();
        }
    }

public:
    std::conditional_t<seg_type == 2, typeE, typeT> get(unsigned int ind){
        ind += sz;
        if constexpr(seg_type == 1) return seg_t[ind];
        for(unsigned int i = lognum; i > 0; i--) push_e(ind >> i);
        if constexpr(seg_type == 2) return seg_e[ind];
        else return seg_t[ind];
    }

    typeT prod(unsigned int l_ind, unsigned int r_ind){
        if constexpr(seg_type == 2) assert(false);
        l_ind += sz, r_ind += sz;

        if constexpr(seg_type == 3){
            for(unsigned int i = lognum; i > 0; i--){
                if(((l_ind >> i) << i) != l_ind) push_e(l_ind >> i);
                if(((r_ind >> i) << i) != r_ind) push_e((r_ind - 1) >> i);
            }
        }

        typeT res_l = e_t(), res_r = e_t();
        while(l_ind < r_ind){
            if(l_ind & 1) res_l = func_tt(res_l, seg_t[l_ind++]);
            if(r_ind & 1) res_r = func_tt(seg_t[--r_ind], res_r);
            l_ind >>= 1, r_ind >>= 1;
        }
        return func_tt(res_l, res_r);
    }

    typeT all_prod(){
        if constexpr(seg_type == 2) assert(false);
        return seg_t[1];
    }

    void apply(unsigned int ind, typeE query){
        ind += sz;
        if constexpr(seg_type != 1){
            for(unsigned int i = lognum; i > 0; i--) push_e(ind >> i);
        }
        seg_t[ind] = func_te(seg_t[ind], query);
        if constexpr(seg_type != 2){
            for(unsigned int i = 1; i <= lognum; i++) update_t(ind >> i);
        }
    }

    void apply(unsigned int l_ind, unsigned int r_ind, typeE query){
        if constexpr(seg_type == 1) assert(false);
        l_ind += sz, r_ind += sz;

        for(unsigned int i = lognum; i > 0; i--){
            if(((l_ind >> i) << i) != l_ind) push_e(l_ind >> i);
            if(((r_ind >> i) << i) != r_ind) push_e((r_ind - 1) >> i);
        }

        unsigned int l2 = l_ind, r2 = r_ind;
        while(l2 < r2){
            if(l2 & 1) apply_e(l2++, query);
            if(r2 & 1) apply_e(--r2, query);
            l2 >>= 1, r2 >>= 1;
        }

        if constexpr(seg_type == 3){
            for(unsigned int i = 1; i <= lognum; i++){
                if(((l_ind >> i) << i) != l_ind) update_t(l_ind >> i);
                if(((r_ind >> i) << i) != r_ind) update_t((r_ind - 1) >> i);
            }
        }
    }

    void _debug(){
        for(unsigned int i = 1; i < sz; i++) push_e(i);
        for(unsigned int i = sz - 1; i > 0; i--) update_t(i);
    }
};

/*
template <typename T>
struct PRODquery_APPLYquery : public segment_tree<T, T,  segtree_type >{ //segtree_type = 1:一点更新区間取得  2:区間更新一点取得  3:区間更新区間取得;
    template <typename ...Args> PRODquery_APPLYquery(Args... args){this->preset(args...);}
    using U = T;
    T e_t(){return 要素モノイドの単位元;（type2の時不要）}
    T e_e(){return 作要素モノイドの単位元;（type1の時不要）}
    T func_tt(T a, T b){return 要素モノイド同士の演算;}（type2の時不要）
    T func_te(T a, E b, int w){return 作用素モノイドの要素モノイドへの作用;}
    E func_ee(E a, E b){return 作用素モノイド同士の演算;}(type1の時不要）)
};
*/

//yuki1558
template <typename S>
struct Rsum_contiguous_subsequencesQ_SUQ : public segment_tree<tuple<S,S,S,S>, tuple<S,S,S,S>, 1>{
    using T = tuple<S,S,S,S>;
    T e_t(){
        T res = {-inf,-inf,-inf,0};
        return res;
    }
    Rsum_contiguous_subsequencesQ_SUQ(int n){this->preset(n,e_t());}
    T func_tt(T a, T b){
        T res;
        auto[ans1,lsum1,rsum1,sum1] = a;
        auto[ans2,lsum2,rsum2,sum2] = b;
        res = make_tuple(max({ans1,ans2,rsum1+lsum2}),max(lsum1,sum1+lsum2),max(rsum2,sum2+rsum1),sum1+sum2);
    return res;}
    T func_te(T a, T b){return b;}
};

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    //cout << fixed << setprecision(15);

    ll N,Q;
    cin >> N >> Q;

    Rsum_contiguous_subsequencesQ_SUQ<ll> Seg(N);
    rep(i,N){
      ll a;
      cin >> a;
      Seg.apply(i,{a,a,a,a});
    }

    while(Q--){
      string s;
      cin >> s;
      if(s=="set"){
        ll i,x;
        cin >> i >> x;
        Seg.apply(i-1,{x,x,x,x});
      }else{
        ll l1,l2,r1,r2;
        cin >> l1 >> l2 >> r1 >> r2;
        l1--;l2--;r1--;r2--;
        if(r1<=l1)r1=l1;
        if(r2<=l2)l2=r2;
        if(l2<r1){
          auto a = get<2>(Seg.prod(l1,l2+1));
          auto b = get<3>(Seg.prod(l2+1,r1));
          auto c = get<1>(Seg.prod(r1,r2+1));
          cout << a+b+c << endl;
        }else{
          auto mid = Seg.prod(r1,l2+1);
          auto l = Seg.prod(l1,r1);
          auto r = Seg.prod(l2+1,r2+1);

          ll res = -inf;
          chmax(res, get<0>(mid));
          chmax(res,get<2>(l)+get<1>(mid));
          chmax(res,get<2>(mid)+get<1>(r));
          chmax(res,get<2>(l)+get<3>(mid)+get<1>(r));
          cout << res << endl;
        }
      }
    }

    //cout << ans << endl;

  return 0;
}
yukicoder

結果

テストケース

ソースコード
