Hdu 2795 线段树的运用

Description
在学校的入口处有一个巨大的矩形广告牌,高为h,宽为w。所有种类的广告都可以贴,比如ACM的广告啊,还有餐厅新出了哪些好吃的,等等。。

在9月1号这天,广告牌是空的,之后广告会被一条一条的依次贴上去。

每张广告都是高度为1宽度为wi的细长的矩形纸条。

贴广告的人总是会优先选择最上面的位置来帖,而且在所有最上面的可能位置中,他会选择最左面的位置,而且不能把已经贴好的广告盖住。

如果没有合适的位置了,那么这张广告就不会被贴了。

现在已知广告牌的尺寸和每张广告的尺寸,求每张广告被贴在的行编号。
Input
多组样例,不超过40个。

对每组样例,第一行包含3个整数h,w,n(1 <= h,w <= 10^9; 1 <= n <= 200,000) -广告牌的尺寸和广告的个数。

下面n行每行一个整数 wi (1 <= wi <= 10^9) - 第i张广告的宽度.
Output
对每张广告,输出它被贴在的行编号(是1到h之间的数),顶部是第一行。如果某广告不能被贴上,则输出-1。
Sample Input

    3 5 5

2
4
3
3
3
Sample Output

1
2
1
3
-1

想法:既然每个广告的高度都为1,那我们可以建立一个长为h的线段树,最底层的每个子节点起初都标记为w,如果放一个广告,则变为w-广告的宽度。而每个父亲节点都维护他的子节点中宽度最长值,这样之后放广告就可以很快判断是放到左子树还是右子树了。

#define _CRT_SECURE_NO_WARNINGS
#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
struct node
{
    //int left, right;      
    int sum;
};
node t[800010];
void creat(int x, int left, int right,int w)
{
    t[x].sum = w;//初始化
    if (left == right)      
        return;
    creat(x * 2, left, (left + right) / 2,w);       
    creat(x * 2 + 1, (left + right) / 2 + 1, right,w);  
}
int add(int x, int left, int right, int q)
{
    int ans;
    if (left == right)      
        {
            t[x].sum -= q; //放一个广告,并进行更新。
            ans = left;
            return ans;
        }
    if (q<=t[x*2].sum) {
    //说明可以左子树中存在一个位置可以放下这个广告。
        ans =  add(x * 2, left, (left + right) / 2, q);
    }
    else
        ans =  add(x * 2 + 1, (left + right) / 2 + 1, right, q);
     //父亲节点进行更新。
        t[x].sum = max(t[x * 2].sum , t[x * 2 + 1].sum);
    return ans;
}
int main(void)
{
    int h,w,n;
    while (scanf("%d%d%d",&h,&w,&n)!=EOF)
    {
        h = min(h,n);
        creat(1, 1, h,w);
        for (int i = 1; i<=n; i++) {
            int a;
            scanf("%d",&a);
            if (a>t[1].sum) {
                printf("-1\n");
            }
            else
                printf("%d\n",add(1, 1, h, a));
        }
    }
    return 0;
}
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页