[线段树] HDU 2795 - Billboard

输入三个整数,h, w, n。代表 h×w 的公告板和接下来有n个公告。
接下来输入n个公告的宽度Ki。高度都为1。
公告的规则是尽量往最上面放置。
问这n个公告分别放置的位置是第几行。
如果不能放置,输出-1。

第一眼看h,w比较大,109, 但是n的最大值是200000。
当h > n的时候,超出的部分是一定不会有公告放置的。
所以就可以用线段树来做了。
构造[1, min(n, h)]的线段树,每个叶子结点代表公告的一行。
线段树的每个结点存储的是左右子树的最大值。
然后不断更新即可。

#include <stdio.h>
#include <algorithm>
#define lson rt << 1
#define rson rt << 1 | 1
using namespace std;
int maxn[200005 << 2];
int h, w, n;
void build(int l, int r, int rt)
{
    maxn[rt] = w;
    if(l == r)
    {
        return ;
    }
    int mid = (l + r) >> 1;
    build(l, mid, lson);
    build(mid + 1, r, rson);
}
void update(int l, int r, int rt, int val, int where)
{
    if(l == r)
    {
        maxn[rt] -= val;
        return ;
    }
    int mid = (l + r) >> 1;
    if(mid >= where) update(l, mid, lson, val, where);
    else update(mid + 1, r, rson, val, where);
    maxn[rt] = max(maxn[lson], maxn[rson]);
}
int query(int l, int r, int rt, int val)
{
    if(l == r && maxn[rt] >= val)
    {
        return l;
    }
    int mid = (l + r) >> 1;
    if(maxn[lson] >= val) return query(l, mid, lson, val);
    else if(maxn[rson] >= val) return query(mid + 1, r, rson, val);
    return -1;
}
int main()
{
    while(~scanf("%d %d %d", &h, &w, &n))
    {
        int k = min(n, h);
        build(1, k, 1);
        for(int i = 0; i < n; i++)
        {
            int temp;
            scanf("%d", &temp);
            int t = query(1, k, 1, temp);
            printf("%d\n", t);
            if(t != -1) update(1, k, 1, temp, t);
        }
    }
    return 0;
}
发布了130 篇原创文章 · 获赞 6 · 访问量 11万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章