Submission #674141

# Submission time Handle Problem Language Result Execution time Memory
674141 2022-12-23T10:12:19 Z DwightKSchrute Growing Trees (BOI11_grow) C++17
10 / 100
140 ms 3024 KB
/*
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
 */
#include<bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

//using namespace __gnu_pbds;
using namespace std;

typedef long double ld;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int>vi;
typedef vector<vector<int>>vvi;
typedef vector<ll>vl;
typedef vector<vl> vvl;
typedef pair<int,int>pi;
typedef pair<ll,ll> pl;
typedef vector<pl> vpl;
typedef vector<ld> vld;
typedef pair<ld,ld> pld;
typedef vector<pi> vpi;

//typedef tree<ll, null_type, less_equal<ll>,rb_tree_tag,tree_order_statistics_node_update> ordered_set;
template<typename T> ostream& operator<<(ostream& os, vector<T>& a){os<<"[";for(int i=0; i<ll(a.size()); i++){os << a[i] << ((i!=ll(a.size()-1)?" ":""));}os << "]\n"; return os;}

#define all(x) x.begin(),x.end()
#define YES out("YES")
#define NO out("NO")
#define out(x){cout << x << "\n"; return;}
#define outfl(x){cout << x << endl;return;}
#define GLHF ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL)
#define print(x){for(auto ait:x) cout << ait << " "; cout << "\n";}
#define pb push_back
#define umap unordered_map


template<typename T>
void read(vector<T>& v){
    int n=v.size();
    for(int i=0; i<n; i++)
        cin >> v[i];
}
template<typename T>
vector<T>UNQ(vector<T>a){
    vector<T>ans;
    for(T t:a)
        if(ans.empty() || t!=ans.back())
            ans.push_back(t);
    return ans;
}



void solve();
int main(){
    GLHF;
    int t=1;
    //cin >> t;
    while(t--)
        solve();
}
struct seg{
    vi t;
    int sz=1;
    seg(int n){
        sz=2*n;
        t.resize(sz);
    }
    void add(int l,int r,int v){
        for(l+=sz/2,r+=sz/2 +1; l<r; l/=2,r/=2){
            if(l&1)t[l++]+=v;
            if(r&1)t[--r]+=v;
        }
    }
    int qur(int i){
        i+=sz/2;
        int ans=0;
        while(i){
            ans+=t[i];
            i/=2;
        }
        return ans;
    }

    int first_greater_equal(int k){
        int l=0,r=sz/2 -1,ans=sz/2;
        while(l<=r){
            int m=(l+r)/2;
            if(qur(m)>=k)
                ans=m,r=m-1;
            else
                l=m+1;
        }
        return ans;
    }
};


void solve() {
    int n,q;
    cin >> n >> q;

    vi a(n);
    read(a);
    sort(all(a));

    seg ST(n);
    for(int i=0; i<n; i++)
        ST.add(i,i,a[i]);

    while(q--){
        char op;
        cin >> op;
        if(op=='C'){
            int l,r;
            cin >> l >> r;

            int first = ST.first_greater_equal(l);
            int last = ST.first_greater_equal(r+1);
            cout << last-first << "\n";

            continue;
        }

        int c,h;
        cin >> c >> h;
        int first = ST.first_greater_equal(h);
        int last = min(first+c-1,n-1);
        int height_last = ST.qur(last);

        last = ST.first_greater_equal(height_last)-1;
        int p_last = last;
        ST.add(first,last,1);
        
        if(last==n-1)
            continue;
        
        c-= last-first+1;

        last = ST.first_greater_equal(height_last+1)-1;
        first = max(p_last+1,last-c+1);
        ST.add(first,last,1);
    }
}
/*
 3 2
1 2 3
F 3 2
C 4 4
 */
# Verdict Execution time Memory Grader output
1 Incorrect 108 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 2 ms 328 KB Output is correct
2 Correct 4 ms 340 KB Output is correct
3 Correct 3 ms 340 KB Output is correct
4 Correct 2 ms 332 KB Output is correct
5 Correct 55 ms 1392 KB Output is correct
6 Incorrect 73 ms 1584 KB Output isn't correct
7 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 71 ms 1556 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 59 ms 1460 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 96 ms 2152 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 120 ms 2224 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 100 ms 2320 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 140 ms 3024 KB Output is correct
2 Incorrect 131 ms 1864 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 126 ms 2680 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 128 ms 2920 KB Output is correct