线段树的运用,意思理解后,代码很好写
题意:在h*w的木板,放1*wi的木块,求能容纳木块的最上边位置
思路:找到最上边的位置后,减去它的宽度。
代码:
#include<cstdio>
#include<iostream>
using namespace std;
int ma[200002<<2];
int n,h,w;
void build(int l,int r,int rt){
ma[rt]=w;
if(l==r){
return;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
}
int query(int x,int l,int r,int rt){
if(l==r){
ma[rt]-=x;
return l;
}
int mid=(l+r)>>1;
int res;
if(ma[rt<<1]>=x)
res=query(x,l,mid,rt<<1);
else
res=query(x,mid+1,r,rt<<1|1);
ma[rt]=max(ma[rt<<1],ma[rt<<1|1]);
return res;
}
int main(){
int wi;
while(scanf("%d%d%d",&h,&w,&n)!=EOF){
if(h>n)
h=n;
build(1,h,1);
for(int i=0;i<n;i++){
scanf("%d",&wi);
if(wi>ma[1]){
printf("-1\n");
}
else
{
printf("%d\n",query(wi,1,h,1));
}
}
}
return 0;
}