hdu2795 Billboard

http://acm.hdu.edu.cn/showproblem.php?pid=2795

线段树，单点更新

好题，查询属于全区间的单点查询，目标值：准备贴的海报宽度

如果左子树的最大值比目标值大，则优先查且只查左子树（我们想要得到编号最小且满足条件的行贴海报）

否则查右子树，如果左右子树的最大值，都比目标值小，就返回-1（没地方贴了～）

WA了一次，没考虑高度为1的展板。

#include <stdio.h>

#define lson l, m, root<<1
#define rson m+1, r, root<<1|1

const int maxn = 200010;
int max1[maxn<<2];

int max(int x, int y)
{
    return x>y? x: y;
}

void push_up(int root)
{
    max1[root] = max(max1[root<<1], max1[root<<1|1]);
}

void build(int x, int l, int r, int root)
{
    int m;
    if(l == r)
    {
        max1[root] = x;
        return;
    }
    m = (l + r) >> 1;
    build(x, lson);
    build(x, rson);
    push_up(root);
}

void update(int x, int sub, int l, int r, int root)
{
    int m;
    if(l == r)
    {
        max1[root] -= sub;
        return;
    }
    m = (l + r) >> 1;
    if(x <= m)
    {
        update(x, sub, lson);
    }
    if(m+1 <= x)
    {
        update(x, sub, rson);
    }
    push_up(root);
}

int query(int x, int l, int r, int root)
{
    int m;
    if(l == r)
    {
        if(max1[root] >= x)
        {
            return l;
        }
        return -1;
    }
    m = (l + r) >> 1;
    if(max1[root<<1] >= x)
    {
        return query(x, lson);
    }
    if(max1[root<<1|1] >= x)
    {
        return query(x, rson);
    }
    return -1;
}

int main()
{
    int n, w, h, a, temp;
    while(~scanf("%d%d%d", &h, &w, &n))
    {
        if(n < h)
        {
            h = n;
        }
        build(w, 1, h, 1);
        while(n-- && scanf("%d", &a))
        {
            temp = query(a, 1, h, 1);
            printf("%d\n", temp);
            if(~temp)
            {
                update(temp, a, 1, h, 1);
            }
        }
    }
    return 0;
}