码迷,mamicode.com
首页 > 其他好文 > 详细

宁波多校(三) G仓鼠的鸡蛋(线段树)

时间:2020-07-07 23:39:52      阅读:91      评论:0      收藏:0      [点我收藏+]

标签:维护   build   ons   res   最大   线段树   isp   print   shu   

用线段树维护区间剩余的最大值,这样可以辅助二分,越左越好

对于用完k堆的,直接将他的值变为0即可

技术图片
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=6e5+10;
struct node{
    int l,r;
    int cnt;
    int k;
}tr[N<<2];
int n,m,k;
int a[N];
void pushup(int u){
    tr[u].cnt=max(tr[u<<1].cnt,tr[u<<1|1].cnt);
}
void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,m,k};
    }
    else{
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
    }
}
void modify(int u,int x,int k){
    if(tr[u].l==tr[u].r){
        tr[u].cnt-=k;
        tr[u].k-=1;
        if(tr[u].k==0){
            tr[u].cnt=0;
        }
        return ;
    }
    int mid=tr[u].l+tr[u].r>>1;
    if(x<=mid)
        modify(u<<1,x,k);
    else{
        modify(u<<1|1,x,k);
    }
    pushup(u);
}
int query(int u,int x){
    if(tr[u].l==tr[u].r){
        return tr[u].l;
    }
    int mid=tr[u].r+tr[u].l>>1;
    int res=0;
    if(tr[u<<1].cnt>=x)
        res=query(u<<1,x);
    else{
        res=query(u<<1|1,x);
    }
    return res;
}
int main(){
    int t;
    cin>>t;
    while(t--){
        cin>>n>>m>>k;
        int i;
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        build(1,1,n);
        for(i=1;i<=n;i++){
            if(a[i]>tr[1].cnt){
                printf("-1\n");
                continue;
            }
            int x=query(1,a[i]);
            printf("%d\n",x);
            modify(1,x,a[i]);
        }
    }
    return 0;
}
View Code

 

宁波多校(三) G仓鼠的鸡蛋(线段树)

标签:维护   build   ons   res   最大   线段树   isp   print   shu   

原文地址:https://www.cnblogs.com/ctyakwf/p/13264113.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!