线段树
题意给一个高为h每行长度为w的广告牌,每次贴长度为Wi高度为1的广告,问该广告能否贴上,如果能,贴在最靠左上的位置并输出第几行,不能则输出-1.
题目数据范围特别大,很容易想到要用线段树来解决问题。开数组时,应想到每条广告都占一行的话,最多只需要20W行,所以开数组大小应为广告的条数与广告牌高度两者的较小值的四倍就行,不需要将数组开到爆。
然后将线段树每个节点剩余宽度初始化为W,每个节点值记录为该节点以下所有叶子节点中的剩余宽度最大值,每次输入广告之后,先判断能否贴上,即与线段树根节点比较,大的话输出-1,小的话肯定能贴上,找到该位置,输出行号,然后更新线段树即可。
由于数据量特别大,注意使用scanf和printf,不然容易超时
#include <iostream>
using namespace std;
int tree[1000000];
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a>b?b:a;
}
void pushup(int rt)
{
tree[rt]=max(tree[rt<<1],tree[rt<<1|1]);
}
/*void build(int l,int r,int rt,int w)
{
tree[rt]=w;
if (l==r)
return ;
int mid=(l+r)/2;
build(l,mid,rt<<1,w);
build(mid+1,r,rt<<1|1,w);
}*/
void update(int w,int l,int r,int rt)
{
if (l==r)
{
tree[rt]-=w;
printf("%d\n",l);
return ;
}
int mid=(l+r)/2;
if (w<=tree[rt<<1])
update(w,l,mid,rt<<1);
else
update(w,mid+1,r,rt<<1|1);
pushup(rt);
}
int main()
{
int h,w,n;
while (scanf("%d%d%d",&h,&w,&n)!=EOF)
{
int s=min(h,n);
//build(1,s,1,w);
for (int i=0;i<=4*s;i++)
tree[i]=w;
for (int i=0;i<n;i++)
{
int a;
scanf("%d",&a);
if (a>tree[1])
printf("-1\n");
else
update(a,1,s,1);
}
}
}