题意是在学校门前有一块宣传板,长w,高h,按照单位长度划分成h行。有n个长度分别为wi(1<=i<=n)的宣传单要贴在上面,每张宣传单的高度都是h,也就是高度占一行,宽度占wi,每张都是贴到最左和最上方的空白处即没贴的地方。每输入一个wi,输出它应该贴在哪一行,如果不能贴就输出-1.
这题用线段树的算法,我第一次没有想到,比较巧妙。把展板旋转九十度,从1到min(h,n)的一行,用每个变量来存下面每一行的长度w,构造线段树,节点上保存下面的长度的最大值。更新的时候就减去相应节点的wi,然后返回那个节点的L或者R,就是应该贴在的那一行。
代码:
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<cstdlib>
#include<math.h>
#define L long long
#define max(a,b) a>b?a:b;
#define min(a,b) a>b?b:a;
using namespace std;
int h,w,n;
int a[400000004];
void build(int l,int r,int n)//构造
{
a[n]=w;
if(l==r)
return ;
int m=(l+r)>>1;
build(l,m,n<<1);
build(m+1,r,n<<1|1);
}
int query(int x,int l,int r,int n)//查询和更新放在一起
{
if(l==r)
{
a[n]-=x;
return l;
}
int m=(l+r)>>1,ret;
if(x<=a[n<<1])//如果x小于左边节点的w最大值,就从左边找
ret=query(x,l,m,n<<1);
else ret = query(x,m+1,r,n<<1|1);//否则就从右边找
a[n]=max(a[n<<1],a[n<<1|1]);//更新节点
return ret;
}
int main()
{
while(~scanf("%d%d%d",&h,&w,&n))
{
int aa;
if(h>n)
h=n;
build(1,h,1);
for(int i=0;i<n;i++)
{
scanf("%d",&aa);
if(aa>a[1])
{
printf("-1\n");
}
else
printf("%d\n",query(aa,1,h,1));
}
}
return 0;
}