HDU 2795 Billboard // 线段树 单点更新

题目描述

HDU 2795 Billboard

解题思路

题目大意:
有一张h×w的矩形告示栏, 和 n 张 1×wi 的海报
海报每次张贴的位置为告示栏剩余位置的左上角.
要求支持下列操作:
1) 对于每张海报 输出其在告示栏的第几行 (若放不下, 则输出 -1)

首先我们知道的是 告示栏上的海报最多只有n张, 即告示栏的行数最多只有n行有效
所以当输入的h大于n时, 令h = n 即可.
线段树每个节点所维护的信息是: 当前区间 所能容纳下的最大宽度

参考代码

#include <stdio.h>
#define lson rt<<1
#define rson rt<<1|1
#define mid ((l+r) >> 1)
inline int max(int a, int b){return a>b?a:b;}
const int MAX_N = 200010;
int num[MAX_N << 2], h, w, n, x;

void pushup(int rt){
    num[rt] = max(num[lson], num[rson]);
}

void build(int l ,int r, int rt){
    num[rt] = w;
    if (l == r) return ;
    build(l, mid, lson);
    build(mid+1, r, rson);
}

int query(int x, int l, int r, int rt){
    if (l == r){
        num[rt] -= x;
        return l;
    }
    // 因为要放在走上角 所以先判断左儿子是否满足条件
    int ans = (num[lson] >= x ? query(x, l, mid, lson) : query(x, mid+1, r, rson)); 
    pushup(rt);
    return ans;
}

int main(){
    while (~scanf("%d %d %d", &h, &w, &n)){
        if (h > n)  h = n;
        build(1, n, 1);
        while (n--){
            scanf("%d", &x);
            if (num[1] < x) puts("-1"); // num[1]代表区间 [1, h] 所能容纳的最大宽度
            else    printf("%d\n", query(x, 1, h, 1));
        }
    }
    return 0;
}