HDU 2795 Billboard (线段树单点更新 && 求区间最值位置)

#include <cstdio>
#include <algorithm>
#include <string.h>
using namespace std;
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
const int maxn = 222222;
int maxv[maxn<<2];
int h, w, n;
void PushUP(int rt) { maxv[rt] = max(maxv[rt<<1] , maxv[rt<<1|1]); }

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

void update(int p,int sc,int l,int r,int rt) {
    if (l == r) {
        maxv[rt] += sc;
        return ;
    }
    int m = l + ((r - l)>>1);
    if (p <= m) update(p , sc , lson);
    else update(p , sc , rson);
    PushUP(rt);
}
int val;
int query(int l,int r,int rt) {
    if(l == r) return l;
    int m = (l + r) >> 1;
    int ret = 0;
    if(maxv[rt<<1] >= val) ret = query(lson);///先判断左孩子，使得贴的位置会尽量靠上
    else ret = query(rson);
    return ret;
}

int main(void)
{
    while(~scanf("%d %d %d", &h, &w, &n)){
        if(h > n) h = n;///这里需要注意 h 最大是 1e9，但是很明显公告占行数最大只会到 n 这么大
        build(1, h, 1);///建树，一开始每一个节点的最大值都是 w
        for(int i=1; i<=n; i++){
            scanf("%d", &val);///输入每张公告的长度
            if(maxv[1] < val) puts("-1");///如果总区间中含有的最大空间都无法容纳当前公告，输出 -1
            else{
                int pos = query(1, h, 1);///查找最大值的位置，注意返回的是位置
                update(pos, -val, 1, h, 1);///将这个位置减去 wi
                printf("%d\n", pos);///输出对应的位置
            }
        }
    }
    return 0;
}