HDU 2795 线段树第三发!

海报贴的优先度:

① 高 > 低 ||  高相等 && 左 > 右

朴素的想法:两层for循环,每贴一张海报考察从第1行到第 min(h,n) 行有没有空余的位置能够贴的下,若贴的下,- = w ;这样的复杂度为n^2,(2*1e5)^2,超时!

学习了线段树之后,很自然的想到,将考察操作用线段树来完成,这样就缩减到了nlogn;

线段树中每个节点保存其左右孩子的可用宽度的最大值,初始都为w;

观察二叉树的性质,如果 seg[i].value>w,即树中是存在满足贴海报的条件的,要按照 条件① 贴海报,当然是不停的先递归其左子树,左子树不满足再递归右子树。

这里的updata操作可以合并到query操作里面(因为query操作每次都是查询到最底层,updata也是从最底层开始),即在回溯的时候更新父亲节点的max值。

代码如下

<span style="font-size:12px;">#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
#define maxn 200005

int a[maxn];
int h,w,n;
struct node
{
    int left,right;
    int value;
}seg[4*maxn];

void build(int i,int l,int r)
{
    seg[i].value=w;
    seg[i].left=l;
    seg[i].right=r;
    if(seg[i].left==seg[i].right) return ;
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
}

int query(int i,int x)
{
    if(seg[i].left==seg[i].right)
    {
        seg[i].value-=x;
        return seg[i].left;
    }
    int ret;
    if(seg[i<<1].value >= x)
        ret=query(i<<1,x);
    else
        ret=query(i<<1|1,x);
    seg[i].value=max(seg[i<<1].value,seg[i<<1|1].value);
    return ret;
}
int main()
{
    int x;
    while(~scanf("%d%d%d",&h,&w,&n))
    {
        build(1,1,min(n,h));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            if(seg[1].value < x)
                printf("-1\n");
            else
                printf("%d\n",query(1,x));
        }
    }
    return 0;
}
</span>


你可能感兴趣的:(HDU 2795 线段树第三发!)