Submission #1131050

#	Time	Username	Problem	Language	Result	Execution time	Memory
1131050		vladilius	Hedgehog Daniyar and Algorithms (IZhO19_sortbooks)	C++20	100 / 100	1542 ms	213952 KiB

sortbooks

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int, int>;
#define pb push_back
#define ff first
#define ss second
#define arr3 array<int, 3>

struct ST{
    vector<int> t;
    int n;
    ST(int ns){
        n = ns;
        t.resize(4 * n);
    }
    void upd(int v, int tl, int tr, int& p, int& x){
        if (tl == tr){
            t[v] = x;
            return;
        }
        int tm = (tl + tr) / 2, vv = 2 * v;
        if (p <= tm){
            upd(vv, tl, tm, p, x);
        }
        else {
            upd(vv + 1, tm + 1, tr, p, x);
        }
        t[v] = max(t[vv], t[vv + 1]);
    }
    void upd(int p, int x){
        upd(1, 1, n, p, x);
    }
    int get(int v, int tl, int tr, int& l, int& r){
        if (l > tr || r < tl) return 0;
        if (l <= tl && tr <= r) return t[v];
        int tm = (tl + tr) / 2, vv = 2 * v;
        return max(get(vv, tl, tm, l, r), get(vv + 1, tm + 1, tr, l, r));
    }
    int get(int l, int r){
        return get(1, 1, n, l, r);
    }
};

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    
    int n, q; cin>>n>>q;
    vector<int> a(n + 2);
    for (int i = 1; i <= n; i++){
        cin>>a[i];
    }
    
    vector<int> log(n + 1);
    for (int i = 2; i <= n; i++) log[i] = log[i / 2] + 1;
    
    const int lg = log[n];
    vector<vector<int>> sp(n + 1, vector<int>(lg + 1));
    for (int i = 1; i <= n; i++) sp[i][0] = a[i];
    for (int j = 1; j <= lg; j++){
        for (int i = 1; i + (1 << j) <= n + 1; i++){
            sp[i][j] = max(sp[i][j - 1], sp[i + (1 << (j - 1))][j - 1]);
        }
    }
    
    auto get = [&](int l, int r){
        if (l > r) return 0;
        int k = log[r - l + 1];
        return max(sp[l][k], sp[r - (1 << k) + 1][k]);
    };
    
    vector<arr3> end[n + 1];
    for (int i = 1; i <= q; i++){
        int l, r, k; cin>>l>>r>>k;
        end[l].pb({r, k, i});
    }
    
    deque<int> d1;
    a[n + 1] = 1e9 + 1;
    d1.pb(n + 1);
    ST T(n);
    deque<int> :: iterator it;
    vector<bool> out(q + 1);
    for (int i = n; i > 0; i--){
        while (!d1.empty() && a[d1.front()] < a[i]){
            T.upd(d1.front(), 0);
            d1.pop_front();
        }
        int j = d1.front(), v = get(i + 1, j - 1);
        d1.push_front(i);
        if (v) T.upd(i, a[i] + v);
        for (auto [r, k, ii]: end[i]){
            it = upper_bound(d1.begin(), d1.end(), r);
            it--;
            j = *it;
            v = T.get(1, j - 1);
            if (j < r){
                v = max(v, a[j] + get(j + 1, r));
            }
            out[ii] = (v <= k);
        }
    }
    
    for (int i = 1; i <= q; i++) cout<<out[i]<<"\n";
}

• Subtask 1: 8 / 8
• Subtask 2: 9 / 9
• Subtask 3: 13 / 13
• Subtask 4: 17 / 17
• Subtask 5: 30 / 30
• Subtask 6: 23 / 23