hdu 2795(线段树:query:区间求最大值的位子)

题意: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 里做了 )