CSES – Hotel Queries

#include<bits/stdc++.h>
#define int long long
#define pii pair<int,int>
#define f first
#define s second
#define fastio ios_base::sync_with_stdio(0);cin.tie(0)
using namespace std;
//declare
const int maxn = 2e5+5;
int n,q;
vector<int> v;
struct seg{
    int L,R;
    int data;
    int lson, rson;
} ST[2*maxn];
int stptr = 1;
//
void build(int L,int R,int idx){
    ST[idx].L = L, ST[idx].R = R;
    if(L+1 == R){
        ST[idx].data = v[L];
        return;
    }
    int mid = (L+R) / 2;
    int lson = ST[idx].lson = stptr++;
    int rson = ST[idx].rson = stptr++;
    build(L,mid,lson), build(mid,R,rson);
    ST[idx].data = max(ST[lson].data, ST[rson].data);
}
void modify(int pos,int x,int idx){
    if(ST[idx].L + 1 == ST[idx].R){
        ST[idx].data += x;
        return;
    }
    int mid = (ST[idx].L + ST[idx].R) / 2;
    int lson = ST[idx].lson, rson = ST[idx].rson;
    if(pos < mid) modify(pos,x,lson);
    else modify(pos,x,rson);
    ST[idx].data = max(ST[lson].data, ST[rson].data);
}
int query(int x,int idx){
    if(ST[idx].L + 1 == ST[idx].R){
        return ST[idx].L;
    }
    int mid = (ST[idx].L + ST[idx].R) / 2;
    int lson = ST[idx].lson, rson = ST[idx].rson;
    if(ST[lson].data >= x) return query(x,lson);
    else return query(x,rson);
}
signed main(){
    fastio;
    cin>>n>>q;
    for(int i=0;i<n;i++){
        int x; cin>>x;
        v.push_back(x);
    }
    build(0,n,0);
    while(q--){
        int x; cin>>x;
        if(ST[0].data < x){
            cout<<0<<" ";
            continue;
        }
        int pos = query(x,0);
        cout<<pos+1<<" ";
        modify(pos,-x,0);
    }
    cout<<"\n";
    return 0;
}