本题的难点在于将题目中的信息转化到线段树中,本题是要找出背景板上可以放下一条公告的第一行的行号。若该行所剩空间大于公告所需空间,则可以放下该公告。那么我们就可以将背景板上每一行所剩的空间当做数组的元素,也就是线段树的叶子结点。而线段树中非叶子结点所储存的就是该区间内所剩空间的最大值,若这个最大值小于公告所需的空间,则说明这段区间内的所有行都放不下这条公告。明白这个转化过程后写个线段树的建立和插入函数就可以啦。本题的一些细节详见注释。
代码如下:
#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=200005;
int ans[maxn<<2];
int h,w,n;
int Max(int a,int b)
{
return a>b?a:b;
}
//初始的时候所有行的最大值都是w,则每段区间的最大值也是w。
void build(int root,int l,int r)
{
ans[root]=w;
if(l==r)
return;
int mid=(l+r)>>1;
build(root<<1,l,mid);
build(root<<1|1,mid+1,r);
}
void insert(int root,int l,int r,int val)
{
if(l==r)
{
printf("%d\n",l);
ans[root]-=val;
return;
}
int mid=(l+r)>>1;
//调用这个函数就代表着背景板中肯定有一行可以放的下当前公告
//所以如果当前区间的左子区间放不下的话,右子区间肯定能放下
if(ans[root<<1]>=val)
insert(root<<1,l,mid,val);
else
insert(root<<1|1,mid+1,r,val);
//插入一条公告后及时更新非叶子结点的值
ans[root]=Max(ans[root<<1],ans[root<<1|1]);
}
int main()
{
while(scanf("%d%d%d",&h,&w,&n)!=EOF)
{
int num;
//若h>n,就将背景板后面用不上的(h-n)行舍掉。而n的范围比h小,所以我是按n的范围来定义线段树数组大小的
if(h>n)
h=n;
build(1,1,h);
for(int i=1;i<=n;i++)
{
scanf("%d",&num);
//若整个背景板都没有比公告大的空闲空间,则输出-1
if(num>ans[1])
printf("-1\n");
else
insert(1,1,h,num);
}
}
return 0;
}