HDU2795 Billboard

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

 

 

#include<cstdio>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn = 222222;
int MAX[maxn << 2];
int h, w, n;
void pushup(int rt)
{
    MAX[rt] = max(MAX[rt << 1], MAX[rt << 1 | 1]);
}
void build(int l, int r, int rt)
{
    MAX[rt] = w;
    if (l == r) return;
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
}
int query(int p, int l, int r, int rt)//这里由于query和update都是对单点操作，所以将两个函数合并为一个了
{
    if (l == r){
        MAX[rt] -= p;
        return l;
    }
    int m = (l + r) >> 1;
    int ret = (MAX[rt<<1] >= p ? query(p, lson) : query(p, rson));//这里判断左子节点是否还有相应的空间，若没有则递归右子节点。
    pushup(rt);
    return ret;
}
int main()
{
    while (~scanf("%d%d%d", &h, &w, &n)){
        if (h > n) h = n;//通知小于h时可以直接缩小h的范围
        build(1, h, 1);
        while(n--){
            int p;
            scanf("%d", &p);
            if (MAX[1] < p) puts("-1");
            else printf("%d\n",query(p, 1, h, 1));
        }
    }
    return 0;
}