Submission #546399

#	Time	Username	Problem	Language	Result	Execution time	Memory
546399		Soul234	Growing Trees (BOI11_grow)	C++14	100 / 100	201 ms	4612 KiB

grow

#include<bits/stdc++.h>
using namespace std;

void DBG() { cerr << "]\n"; }
template<class H, class... T> void DBG(H h, T... t) {
    cerr << h; if(sizeof...(t)) cerr << ", ";
    DBG(t...);
}
#ifdef LOCAL
#define dbg(...) cerr << "[" << #__VA_ARGS__ << "]: [", DBG(__VA_ARGS__)
#else
#define dbg(...) 0
#endif // LOCAL

#define FOR(i,a,b) for(int i = (a) ; i<(b) ; i++)
#define F0R(i,a) FOR(i,0,a)
#define ROF(i,a,b) for(int i = (b)-1 ; i>=(a) ; i--)
#define R0F(i,a) ROF(i,0,a)
#define each(e,a) for(auto &e : (a))
#define sz(v) (int)(v).size()
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define pb push_back
#define tcT template<class T
#define nl "\n"

using ll = long long;
using vi = vector<int>;
using pi = pair<int,int>;
using str = string;
tcT> using V = vector<T>;
tcT> using pqg = priority_queue<T,vector<T>,greater<T>>;

void setIO(string NAME = "") {
    cin.tie(0)->sync_with_stdio(0);
    if(sz(NAME)) {
        freopen((NAME + ".inp").c_str(),"r",stdin);
        freopen((NAME + ".out").c_str(),"w",stdout);
    }
}

const int SZ = 1<<17;
int st[SZ<<1], lzy[SZ<<1], H[SZ];
int N, M;

void build(int ind = 1, int L = 0, int R = N) {
    if(L+1 == R) st[ind] = H[L];
    else {
        int mid = (L + R) >> 1;
        build(ind<<1, L, mid);
        build(ind<<1|1, mid, R);
        st[ind] = max(st[ind<<1], st[ind<<1|1]);
    }
}

void addTo(int ind, int val) {
    st[ind] += val;
    lzy[ind] += val;
}

void push(int ind) {
    addTo(ind<<1, lzy[ind]);
    addTo(ind<<1|1, lzy[ind]);
    lzy[ind] = 0;
}

void upd(int l, int r, int ind = 1, int L = 0, int R = N) {
    if(L >= r || R <= l) return;
    if(l <= L && R <= r) return addTo(ind, 1);
    push(ind);
    int mid = (L + R) >> 1;
    upd(l, r, ind<<1, L, mid);
    upd(l, r, ind<<1|1, mid, R);
    st[ind] = max(st[ind<<1], st[ind<<1|1]);
}

int getVal(int id, int ind = 1, int L = 0, int R = N) {
    if(L+1 == R) return st[ind];
    else {
        push(ind);
        int mid = (L + R) >> 1;
        if(id>=mid) return getVal(id, ind<<1|1, mid, R);
        else return getVal(id, ind<<1, L, mid);
    }
}

int query(int val, int ind = 1, int L = 0, int R = N) {
    if(L+1 == R) {
        if(st[ind] > val) return L;
        else return R;
    }
    push(ind);
    int mid = (L + R) >> 1;
    if(st[ind<<1] > val) return query(val, ind<<1, L, mid);
    else return query(val, ind<<1|1, mid, R);
}

void solve() {
    cin>>N>>M;
    F0R(i,N) cin>>H[i];
    sort(H, H+N);
    build();
    F0R(_,M) {
        char t;
        cin>>t;
        if(t == 'F') {
            int amt, mnH;
            cin>>amt>>mnH;
            int lo = query(mnH-1);
            if(lo == N) continue;
            int hi = min(N-1, lo + amt -1);
            int val = getVal(hi);
            int hi_ = query(val-1);
            if(lo<hi_) upd(lo,hi_);
            amt = hi - hi_ + 1;
            hi_ = query(val);
            upd(hi_- amt, hi_);
        }
        else {
            int lo, hi; cin>>lo>>hi;
            cout << query(hi) - query(lo-1) << nl;
        }
    }
}

int main() {
    setIO();

    int t=1;
    //cin>>t;
    while(t-->0) {
        solve();
    }

    return 0;
}

Compilation message (stderr)

grow.cpp: In function 'void setIO(std::string)':
grow.cpp:37:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   37 |         freopen((NAME + ".inp").c_str(),"r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
grow.cpp:38:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   38 |         freopen((NAME + ".out").c_str(),"w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10