[線段樹] 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萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章