HDU2795 Billboard(线段树)

题目链接：HDU2795.Billboard

在一个h * w的板子上贴n张海报，每张海报的面积是1 * wi，依次张贴。

要求尽量往上贴，同一行尽量往左贴，输出如果无法贴下输出-1。注意的是如果h过大的情况，因为最多20w张海报，所以如果h大于20w，就构建20w个叶子节点即可。边输入边更新即可。

稍微改了下自己的代码格式。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
#define lson l, mid, t<<1
#define rson mid+1, r, t<<1|1
#define MAX 200010
#define LL(x) (x << 1)
#define RR(x) (x << 1 | 1)
#define MID(x, y) (x + ((y - x) >> 1))
int a[MAX];
int w, h, n;
class segment_tree {
public :
int tree[MAX * 4];
void pushup(int t) {
tree[t] = w;
}
void build(int l, int r, int t) {
if(l == r) {
tree[t] = w; return;
}
int mid = MID(l, r);
build(lson), build(rson);
pushup(t);
}
int update(int c, int l, int r, int t) {
if(l == r) {
tree[t] -= c; return l;
}
else {
int ret;
int mid = MID(l, r);
if(tree[LL(t)] >= c) ret = update(c, lson);
else ret = update(c, rson);
tree[t] = max(tree[LL(t)], tree[RR(t)]);
return ret;
}
}
}seg;
int main() {
while(~scanf("%d%d%d", &h, &w, &n)) {
seg.build(1, min(h, MAX), 1);
for(int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
if(seg.tree[1] < a[i]) printf("-1\n");
else printf("%d\n", seg.update(a[i], 1, min(h, MAX), 1));
}
}
return 0;
}