[线段树] HDU 2795 - Billboard

来源:互联网 发布:cpi ppi数据 编辑:程序博客网 时间:2024/05/05 14:45

输入三个整数,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 | 1using 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;}
0 0
原创粉丝点击