题意:有一块尺寸为h*w的矩形长板,要在上面贴1*wi的海报n张,选择贴海报的位置是:尽量高,同一高度,选择尽量靠左的地方。要求输出每张海报的高度位置。
思路:因为海报的宽度都是1,就可以把长板旋转90°(长变成高,高变成长),把底层的每张海报当作单个结点(线段树中 tree[root].l==tree[root].r的点)用来表示高度。
因为只有200000个点,所以线段树只用开200000就够了
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#define N 200005
using namespace std;
struct Tree{
int l, r, sum, add,num;//num记录这个区间内可以贴的海报的最长长度
}tree[N<<2];
int a;
int h,w,n;
//延迟更新
void pushup(int root){//儿子把信息传递给父亲
//if(tree[root].l==tree[root].r)return ;
tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum; //<<1:*2; <<1|1:*2+1
tree[root].num=max(tree[root<<1].num,tree[root<<1|1].num);
return ;
}
void pushdown(int root){//父亲把自己的信息传给儿子
if(tree[root].l==tree[root].r)return ;
if(tree[root].add==-1)return ;
tree[root<<1].add=tree[root<<1|1].add=tree[root].add;
tree[root<<1].sum+=(tree[root<<1].r-tree[root<<1].l+1)*tree[root].add; //长度*父亲.add
tree[root<<1|1].sum+=(tree[root<<1|1].r-tree[root<<1|1].l+1)*tree[root].add;
tree[root].add=-1;
return ;
}
//初始化
void build(int l,int r,int root){
tree[root].l=l;
tree[root].r=r;
tree[root].sum=0;
tree[root].add=-1;
if(l==r){//直到单个区间位置
tree[root].sum=1;
tree[root].num=w;//开始时有w的长度可以用来贴海报
return;
}
int mid=(l+r)>>1;
build(l,mid,root<<1);//左区间
build(mid+1,r,root<<1|1); //右区间
pushup(root);//把儿子的信息更新到父亲
return;
}
//更新区间
void update(int l,int r,int z,int root){
if(l==tree[root].l&&tree[root].r==r){
//tree[root].sum+=(tree[root].r-tree[root].l+1)*z;
tree[root].num-=z;
tree[root].add=z;//把要更新的内容保存下来,等到要用儿子时 再去更新
return ;
}
//pushdown(root);//用父亲的信息更新儿子
int mid=tree[root].l+tree[root].r>>1;
if(r<=mid)update(l,r,z,root<<1);
else if(l>mid)update(l,r,z,root<<1|1);
else {
update(l,mid,z,root<<1);
update(mid+1,r,z,root<<1|1);
}
pushup(root);//更新父亲
return ;
}
//查找区间信息
int query(int a,int root){
if(tree[root].num<a)
return -1;
/*如果整个长板能贴的最长长度都比这张海报的长度小,说明这张海报无法贴下*/
if(tree[root].num>=a&&tree[root].l==tree[root].r)
return tree[root].l;
//pushdown(root);
int mid=tree[root].l+tree[root].r>>1;
if(tree[root<<1].num>=a)return query(a,root<<1);//先从左边贴起
else if(tree[root<<1|1].num>=a)return query(a,root<<1|1);
}
int main()
{
while(~scanf("%d%d%d",&h,&w,&n))
{
build(1,n,1);
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
int ans=query(a,1);
if(ans>h)ans=-1;//如果得到的高度大于长板本身的高度说明答案不存在
printf("%d\n",ans);
if(ans!=-1)
{
update(ans,ans,a,1);
}
}
}
return 0;
}