oj.uz

Submission #1097170

# Submission time Handle Problem Language Result Execution time Memory
1097170 2024-10-06T11:06:04 Z anhthi Growing Trees (BOI11_grow) C++14
100 / 100
 93 ms 6220 KB 
#include <bits/stdc++.h>
 
using namespace std;
 
#define fi first
#define se second
#define ll long long
#define mp(x, y) make_pair(x, y)
#define sz(v) ((int) (v).size())
#define all(v) (v).begin(), (v).end()
#define MASK(i) (1LL << (i))
#define BIT(x, y) (((x) >> (y)) & 1)
#define pb push_back
#define max_rng *max_element
#define min_rng *min_element
#define rep(i, n) for(int i = 1, _n = (n); i <= _n; ++i)
#define forn(i, a, b) for(int i = (a), _b = (b); i <= _b; ++i)
#define ford(i, a, b) for(int i = (a), _b = (b); i >= _b; --i)
 
template <class X, class Y>
    inline bool maximize(X &x, Y y) {
        return (x < y ? x = y, true : false);
    }
 
template <class X, class Y>
    inline bool minimize(X &x, Y y) {
        return (x > y ? x = y, true : false);
    }
 
template <class X>
    inline void compress(vector<X> &a) {
        sort(all(a));
        a.resize(unique(all(a)) - a.begin());
    }
 
int ctz(ll x) { return __builtin_ctzll(x); }
int lg(ll x) { return 63 - __builtin_clzll(x); }
int popcount(ll x) { return __builtin_popcount(x); }
 
mt19937_64 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
ll rand(ll l, ll r) {
    return l + abs((ll) rng()) % (r - l + 1);
}
 
const ll oo = (ll) 1e17;
const int inf = (ll) 1e9 + 7, mod = (ll) 1e9 + 7;
 
const int N = 2e6 + 5, M = 2e5 + 5, BASE = 307, LOG = 23;
 
void add(int &x, int y) { x += y; if (x >= mod) x -= mod; }
void sub(int &x, int y) { x -= y; if (x < 0) x += mod; }
    
int n, m;
int a[N];
 
struct Seg {
 
    int n;
    vector<int> st, lz;
 
    Seg(int n) : n(n) {
        st.resize(n << 2);
        lz.resize(n << 2);
        build(1, 1, n);
    }
 
    void build(int i, int l, int r) {
        if (l == r) return st[i] = a[l], void();
        int m = (l + r) >> 1;
        build(2 * i, l, m);
        build(2 * i + 1, m + 1, r);
 
        st[i] = max(st[2 * i], st[2 * i + 1]);
    }
 
    void setNode(int i, int x) {
        st[i] += x;
        lz[i] += x;
    }
    void down(int i) {
        if (!lz[i]) return;
        setNode(2 * i, lz[i]);
        setNode(2 * i + 1, lz[i]);
        lz[i] = 0;
        return;
    }
 
    void upd(int i, int l, int r, int u, int v, int x) {
        if (l > v || r < u) return;
        if (l >= u && r <= v) return setNode(i, x);
        down(i);
        int m = (l + r) >> 1;
        upd(2 * i, l, m, u, v, x);
        upd(2 * i + 1, m + 1, r, u, v, x);
        st[i] = max(st[2 * i], st[2 * i + 1]);
    }
 
    void upd(int u, int v, int x) {
        if (u > v) return;
        return upd(1, 1, n, u, v, x);
    }
 
    int get(int p) {
        int i = 1;
        for (int l = 1, r = n; l < r; ) {
            down(i);
            int m = (l + r) >> 1;
            if (p > m) {
                l = m + 1;
                i = 2 * i + 1;
            }
            else {
                r = m;
                i = 2 * i;
            }
        }
        return st[i];
    }
 
    int walk(int Max) {
        if (st[1] < Max) return n + 1;
 
        int i = 1;
        int l = 1, r = n;
        while (true) {
            if (l == r) return l;
            down(i);
            int m = (l + r) >> 1;
            if (st[2 * i] >= Max) {
                r = m;
                i = 2 * i;
            }
            else {
                l = m + 1;
                i = 2 * i + 1;
            }
        }
    }
 
};
 
int main(void) {
 
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
 
    cin >> n >> m;
    rep(i, n) cin >> a[i];
    sort(a + 1, a + 1 + n);
 
    Seg s(n);
 
    rep(_, m) {
        char type;
        cin >> type;
 
        if (type == 'F') {
            int c, h;
            cin >> c >> h;
 
            int pos = s.walk(h);
            if (pos + c - 1 >= n) {
                s.upd(pos, n, +1);
                continue;
            }
            int cur = s.get(pos + c - 1);
            int fst = max(pos, s.walk(cur)), lst = s.walk(cur + 1) - 1;
            int cnt = pos + c - fst;
            s.upd(pos, fst - 1, +1);
            s.upd(lst - cnt + 1, lst, +1);
        }   
        else {
            int l, r;
            cin >> l >> r;
            cout << s.walk(r + 1) - s.walk(l) << '\n';
        }
 
    }
 
    return 0;
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 46 ms 4956 KB Output is correct
2 Correct 78 ms 5204 KB Output is correct
3 Correct 42 ms 4944 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 2 ms 348 KB Output is correct
4 Correct 1 ms 480 KB Output is correct
5 Correct 28 ms 1620 KB Output is correct
6 Correct 29 ms 1856 KB Output is correct
7 Correct 3 ms 600 KB Output is correct
8 Correct 14 ms 1368 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 1884 KB Output is correct
2 Correct 33 ms 1872 KB Output is correct
3 Correct 2 ms 604 KB Output is correct
4 Correct 25 ms 1628 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 2152 KB Output is correct
2 Correct 31 ms 2012 KB Output is correct
3 Correct 6 ms 604 KB Output is correct
4 Correct 31 ms 1884 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 40 ms 3916 KB Output is correct
2 Correct 68 ms 4176 KB Output is correct
3 Correct 9 ms 1368 KB Output is correct
4 Correct 33 ms 4268 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 59 ms 4116 KB Output is correct
2 Correct 64 ms 4688 KB Output is correct
3 Correct 38 ms 4332 KB Output is correct
4 Correct 9 ms 1372 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 40 ms 4440 KB Output is correct
2 Correct 46 ms 4692 KB Output is correct
3 Correct 36 ms 4944 KB Output is correct
4 Correct 9 ms 1368 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 74 ms 5008 KB Output is correct
2 Correct 67 ms 4324 KB Output is correct
3 Correct 19 ms 4696 KB Output is correct
4 Correct 48 ms 4792 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 50 ms 4944 KB Output is correct
2 Correct 66 ms 5220 KB Output is correct
3 Correct 93 ms 5716 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 51 ms 6220 KB Output is correct