HDU_2795 Billboard 【线段树】

原题链接

题目意思：

给定一个公告板，让你贴公告，公告是尽量往左上角贴， 如果可以贴，输出行号，如果不能贴，输出-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)//让所有子树的根的最大宽度都为width
{
    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)) //换做cin会超时
    {
        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//进行查找并输出ans
            {
                int ans = query(w,1,height,1);
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}