HDU 2795 - Billboard(线段树)

题目链接：

　　http://acm.hdu.edu.cn/showproblem.php?pid=2795

题目大意：

　　给定一个h * w大小的黑板报，h为黑板报的高，w为黑板报的宽。现在要贴上很多张海报，每张海报的大小为1 * m，m就是输入的海报宽度。从左上贴起，每次从上到下看有没有足够的空间贴海报，有就贴上。给定一组输入表示海报的宽度，输出每个海报贴的高度位置，若没得贴就输出-1。

解题思路：

　　一开始根本不知道怎么用线段树，看了题解之后才知道其模型其实很简单，构建一个大小等于海报数量的线段树，即表明每张海报贴一行就是了。然后维护的就是线段树左右子树的最大长度，这也很好理解，比如要贴的海报大小为a，如果线段树第一个节点值大于等于a，那么说明在线段树中有那么一个节点的剩余宽度大于a，也就是这一行可以贴这张海报。如果线段树第一个节点值小于a，那么说明这个黑板报的最大剩余宽度不足a，明显没法贴了。所以其实分析之后就是一个维护左右子节点最大值的线段树而已。

代码：

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
#include <algorithm>
using namespace std;

#define ll long long
#define INF 0x3f3f3f3f
#define LL_INF 0x3f3f3f3f3f3f3f3f
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1|1
#define MAX 200200

int h, w;
int maxn[MAX << 2];

void pushUp(int rt)
{
    maxn[rt] = max(maxn[rt << 1], maxn[rt << 1|1]);
}

void build(int l, int r, int rt)
{
    if (l == r) {
        maxn[rt] = w;
        return;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    pushUp(rt);
}

int query(int c, int l, int r, int rt)
{
    if (l == r) {
        maxn[rt] -= c;
        return l;
    }
    int m = (l + r) >> 1;
    int res = maxn[rt << 1] >= c ? query(c, lson) : query(c, rson);
    pushUp(rt);
    return res;
}

int main()
{
    //freopen("debug\\in.txt", "r", stdin);
    //freopen("CON", "w", stdout);
    int i, j, k;
    int n;
    while (~scanf("%d%d%d", &h, &w, &n)) {
        int right;
        if (h > n) right = n;
        else right = h;
        build(1, right, 1);
        int width;
        while (n--) {
            scanf("%d", &width);
            if (maxn[1] < width)
                printf("-1\n");
            else
                printf("%d\n", query(width, 1, right, 1));
        }
    }
    return 0;
}