hdu 2795 (线段树)

原题在此


题意是在学校门前有一块宣传板,长w,高h,按照单位长度划分成h行。有n个长度分别为wi(1<=i<=n)的宣传单要贴在上面,每张宣传单的高度都是h,也就是高度占一行,宽度占wi,每张都是贴到最左和最上方的空白处即没贴的地方。每输入一个wi,输出它应该贴在哪一行,如果不能贴就输出-1.


这题用线段树的算法,我第一次没有想到,比较巧妙。把展板旋转九十度,从1到min(h,n)的一行,用每个变量来存下面每一行的长度w,构造线段树,节点上保存下面的长度的最大值。更新的时候就减去相应节点的wi,然后返回那个节点的L或者R,就是应该贴在的那一行。


代码:

#include
#include
#include
#include
#include
#include
#include
#define L long long
#define max(a,b) a>b?a:b;
#define min(a,b) a>b?b:a;
using namespace std;

int h,w,n;
int a[400000004];

void build(int l,int r,int n)//构造
{
    a[n]=w;
    if(l==r)
        return ;
    int m=(l+r)>>1;
    build(l,m,n<<1);
    build(m+1,r,n<<1|1);
}


int query(int x,int l,int r,int n)//查询和更新放在一起
{
    if(l==r)
    {
        a[n]-=x;
        return l;
    }
    int m=(l+r)>>1,ret;
    if(x<=a[n<<1])//如果x小于左边节点的w最大值,就从左边找
        ret=query(x,l,m,n<<1);
    else ret = query(x,m+1,r,n<<1|1);//否则就从右边找
    a[n]=max(a[n<<1],a[n<<1|1]);//更新节点
    return ret;
}



int main()
{
    while(~scanf("%d%d%d",&h,&w,&n))
    {
        int aa;
        if(h>n)
            h=n;
        build(1,h,1);
        for(int i=0;ia[1])
            {
                printf("-1\n");
            }
            else
                printf("%d\n",query(aa,1,h,1));
        }
    }
    return 0;
}


你可能感兴趣的:(hdu 2795 (线段树))