HDU 2795 Billboard【线段树】

题目链接

题目意思

有一块h*w的广告牌，现在要往上边贴广告，所贴广告的高度都为1，要求所贴的广告尽量往上和往左贴，如果可以贴，输出你所帖广告的行号，否则输出“-1”。

解题思路

这道题用线段树来解。我们以广告牌的高度作为标准来构造线段树，另外用一个数组来存储线段树每个节点的宽度。贴广告的时候我们从左子树往右子树来贴，这样就是从上往下来贴的。具体过程看代码吧！

代码部分

#include <iostream>
#include <stdio.h>
#define lchild left,mid,root<<1
#define rchild mid+1,right,root<<1|1
using namespace std;

const int maxn=1e8;
int Max[maxn<<2];
int height,width,n;
///构建线段树
void build(int left,int right,int root)
{
Max[root]=width;
if(left==right)
return;
int mid=(left+right)>>1;
build(lchild);
build(rchild);
}
///更新线段树
void push_up(int root)
{
Max[root]=max(Max[root<<1],Max[root<<1|1]);///找出左右子树最大的宽度赋给父节点
}
///查询并插入
int query(int w,int left,int right,int root)
{
if(left==right)
{
Max[root]-=w;
return left;///返回行号
}
int mid=(left+right)>>1;
int ans=0;
if(w<=Max[root<<1])///如果左子树中有大于w的边，就将其插入左子树中
ans=query(w,lchild);
else
ans=query(w,rchild);
push_up(root);
return ans;
}
int main()
{
while(~scanf("%d%d%d",&height,&width,&n))
{
if(n<height)
height=n;
build(1,height,1);
int w;
while(n--)
{
scanf("%d",&w);
if(Max[1]<w)///如果根的最大长度没有w那么长，直接输出-1
printf("-1\n");
else
{
int ans=query(w,1,height,1);
printf("%d\n",ans);
}
}
}
return 0;
}