题意:h*w的木板,放进一些1*L的物品,求每次放空间能容纳且最上边的位子
思路:每次找到最大值的位子,然后减去L
线段树功能:query:区间求最大值的位子(直接把update的操
#include <cstdio>
#include <iostream>
#include <algorithm>
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
using namespace std;
const int maxn = 222222;
int h , w , n;
int MAX[maxn<<2];
void PushUP(int rt) //往上更新
{
MAX[rt] = max(MAX[rt<<1] , MAX[rt<<1|1]);
}
void build(int l,int r,int rt)
{
MAX[rt] = w;
if (l == r) return ;
int m = (l + r) >> 1;
build(lson);
build(rson);
}
int query(int x,int l,int r,int rt)
{
if (l == r) {
MAX[rt] -= x;//找到位置直接减
return l;
}
int m = (l + r) >> 1;
int ret = (MAX[rt<<1] >= x) ? query(x , lson) : query(x , rson);//按照题意,尽量往上更新
PushUP(rt);//更新
return ret;
}
int main()
{
while (~scanf("%d%d%d",&h,&w,&n))
{
if (h > n) h = n;
build(1 , h , 1);//h>n时,保证所有符合要求的都能放得下
while (n --)
{
int x;
scanf("%d",&x);
if (MAX[1] < x) puts("-1");//放不下
else printf("%d\n",query(x , 1 , h , 1));
}
}
return 0;
}
作在 query 里做了 )