题目链接:https://cn.vjudge.net/problem/HDU-2795
题目大意
每年开学,正是各大社团招新之际。
每个社团为了吸引更多的小学妹小学弟就会派出身强体壮的华师男去海报墙上粘贴海报。
开学之初,高为h,宽为w的海报墙还是空的。然后,华师男轮流粘贴高为1,宽为wi的海报。
贴海报时,机智的华师男总是会优先选择最上面的位置来帖,而且在所有最上面的可能位置中,他会选择最左面的位置。但是不能把已经贴好的海报盖住并且不能超出海报墙的范围。
机智的华师男能够自然能够秒秒钟得出自己的海报应该粘贴在第几行啦。
Input
有多组样例,但是不会超过40个。
对于每组样例,第一行包含3个整数h,w,n(1 <= h,w <= 10^9; 1 <= n <= 200,000) -海报墙的高度和宽度,以及海报的张数。
下面n行每行一个整数 wi (1 <= wi <= 10^9) - 代表第i张海报的宽度.
Output
对于每张海报,输出它被贴在第几行,顶部是第一行。如果某广告不能贴不下了,则输出-1。
分析
线段树维护区间最大值。因为最多有2e5张海报,所以可以用线段树来做。查询时整个区间查询,先找左子区间,再找右子区间,而且如果左子区间最大值大于等于wi的话,就只查询左子区间,这样会省时。
代码
#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
#define inf 0x3f3f3f3f
const int N = 200005;
int h,w,n,ans;
struct node{
int l,r,val;
}tr[N<<2];
void pushup(int m)
{
tr[m].val = max(tr[m<<1].val,tr[m<<1|1].val);
}
void build(int m,int l,int r)
{
tr[m].l = l;
tr[m].r = r;
tr[m].val = w;
if(l == r) return ;
int mid = (l + r) >> 1;
build(m<<1,l,mid);
build(m<<1|1,mid+1,r);
}
void updata(int m,int id,int val)
{
if(tr[m].l == id && tr[m].r == id)
{
tr[m].val -= val;
return ;
}
int mid = (tr[m].l + tr[m].r) >> 1;
if(id <= mid)
updata(m<<1,id,val);
else
updata(m<<1|1,id,val);
pushup(m);
}
void query(int m,int l,int r,int w)
{
if(tr[m].val < w) return ;
if(l == r)
{
ans = min(ans,l);//对应海报墙第l行
return ;
}
int mid = (l + r) >> 1;
if(tr[m<<1].val >= w)
query(m<<1,l,mid,w);
else if(tr[m<<1|1].val >= w)
query(m<<1|1,mid+1,r,w);
}
int main()
{
while(~scanf("%d%d%d",&h,&w,&n))
{
int tmp = min(h,n);
build(1,1,tmp);
for(int i=1;i<=n;i++)
{
int w;
scanf("%d",&w);
ans = inf;
query(1,1,tmp,w);
if(ans == inf || ans > h)
printf("-1\n");
else
{
printf("%d\n",ans);
updata(1,ans,w);
}
}
}
return 0;
}