HDU 2795 - Billboard（线段树）

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

【题意】
有一个高 $h$ 宽 $w$ 的公告栏，现在要在上边贴 $N$ 个 $1*W$ 的公告。要求要在能够完全放下的区域贴公告，并且贴公告时遵循最上最左的原则。问你每张公告分别贴在第几行，若公告无法插入输出 $(-1$ $N<=2×10^5 \ \ h,w<=10^9)$

【思路】
按行建立线段树，初始化每个区间最大值 $maxv$为 $w$,之后每次贴公告就是找一个 $maxv$值大于等于当前公告长度且下标尽量小的行插入，之后维护区间最大值。若 $maxv < W$说明不能插入。

#include<bits/stdc++.h>
#define node tree[id]
#define lson tree[id<<1]
#define rson tree[id<<1|1]
using namespace std;

const int maxn=200005;

struct Tree{
	int left,right;
	int maxv;
}tree[maxn<<2];

int h,w,n;

void pushup(int id){node.maxv=max(lson.maxv,rson.maxv);}

void build(int id,int le,int ri){
	node.left=le;
	node.right=ri;
	node.maxv=w;
	if(le==ri) return;
	int mid=(le+ri)>>1;
	build(id<<1,le,mid);
	build(id<<1|1,mid+1,ri);
}

void update(int id,int pos,int val){
	if(node.left==node.right){
		node.maxv+=val;
		return;
	}
	int mid=(node.left+node.right)>>1;
	if(pos<=mid) update(id<<1,pos,val);
	else update(id<<1|1,pos,val);
	pushup(id);
}

int query(int id,int val){
	if(node.left==node.right){
		if(node.maxv>=val) return node.left;
		else return -1;
	}
	if(lson.maxv>=val) return query(id<<1,val);
	else return query(id<<1|1,val);
}

int main(){
	while(scanf("%d%d%d",&h,&w,&n)==3){
		build(1,1,n);
		for(int i=0;i<n;++i){
			int x;
			scanf("%d",&x);
			if(x>w){
				puts("-1");
				continue;
			}
			int ans=query(1,x);
			if(ans!=-1 && ans>h) ans=-1;
			printf("%d\n",ans);
			if(ans!=-1) update(1,ans,-x);
		}
	}
	return 0;
}