单点更新区间求和贴海报问题 hdu2795 billbord

hdu2795  billbord

  题目大意：有一块板子，长宽分别为W，H，然后有n块1*w海报，让你把这n快海报贴在这块板子上，尽量板子的上方贴，同一行尽量往板子的左边贴。对于第i块海报如果能够贴下则输出能贴在第几行，否则输出-1。

    解题思路： 设线段树的操作长度为H，则对应的叶子节点的最大值为W。设每次找到符合要求的位置（板子最上最左，对应线段树中最下最左）减去相应的w值。   

#include <iostream>
#include <cstdio>
using namespace std;

#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=200005;
int tree[4*maxn];
int t_max[4*maxn];
int n, m, max_w, nn;

void build(int u, int l, int r)
{
tree[u]=max_w;
if(l==r)
{
return ;
}
int mid=(l+r)>>1;
build(lz);
build(rz);
}

void Update(int u, int l, int r, int id, int w)
{
if(l==id&&id==r)
{
tree[u]-=w;
return ;
}
int mid=(l+r)>>1;
if(id<=mid)
Update(lz,id,w);
else Update(rz,id,w);
tree[u]=max(tree[2*u],tree[2*u+1]); ///向上更新最大子区间段

}

int Query(int u, int l, int r, int w)
{
if(tree[u]<w) return -1; ///只需要判断此区间最大空余w是否大于w即可，节省好多时间
if(l==r)
{
return l;
}
int mid=(l+r)>>1;
if(tree[2*u]>=w)
return Query(lz,w);
else
return Query(rz,w);
}

int main()
{
while(cin >> n >> max_w >> m)
{
nn=min(n,m);
build(1,1,nn);
while(m--)
{
int w;
scanf("%d",&w);
int id=Query(1,1,nn,w);
if(id==-1) puts("-1");
else
{
printf("%d\n",id);
Update(1,1,nn,id,w);
}
}
}
}