题目:http://acm.hdu.edu.cn/showproblem.php?pid=2795
单点类更新的线段树,按高度h查询即可。直接标记每段的最大可添加宽度。
下面是AC代码:
#include<cstdio>
#include<algorithm>
using namespace std;
#define MAX 200010
struct node{
int id,l,r;
int max; //该区间剩余可以装载的最大值
}T[MAX*4];
int h,w,n;
void build(int id,int l,int r){
T[id].id=id; T[id].l=l; T[id].r=r; T[id].max = w;
if(l<r){
int mid=(l+r)>>1;
build(id<<1,l,mid);
build((id<<1)|1,mid+1,r);
}
}
int query(int id,int val){
if(T[id].l==T[id].r){
if(T[id].max<val) return -1;
T[id].max -= val;
return T[id].l;
}
if(val<=T[id].max){
int x = 0;
if( val <= T[id<<1].max){
x = query(id<<1,val);
}else {
x = query((id<<1)|1,val);
}
T[id].max = max(T[id<<1].max,T[(id<<1)|1].max);
return x;
}else
return -1;
}
int main(){
int wi;
while(scanf("%d%d%d",&h,&w,&n)!=EOF){
if(h>n) h=n;
build(1,1,h);
for(int i=0;i<n;i++){
scanf("%d",&wi);
printf("%d\n",query(1,wi));
}
}
return 0;
}