Hdu 2795

1690 ワード

【題意】
h*wの黒板があって、その中のn人は1*wiの新聞を貼って、貼る時満足してできるだけ上&&左に寄りかかります.
各投稿位置の行番号を出力します.
【問題解きの考え方】
セグメントツリーの区間は[1,h]であり、ツリー上のノードは含まれる区間の最大幅を表し、wに初期化される.
【ACコード】
#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int nn = 200010;
int h,w,n,num;
struct node{
   int l,r;
   int ms;
}Tree[nn<<2];
void Push_Up(int rt)
{
    Tree[rt].ms = max(Tree[rt<<1].ms,Tree[rt<<1|1].ms);
}
void Build(int l,int r,int rt)
{
    Tree[rt].l=l,Tree[rt].r=r;
    Tree[rt].ms=w;
    if(l==r)return ;
    int m =(l+r)>>1;
    Build(lson);
    Build(rson);
}
void Update(int x,int rt)
{
    if(Tree[rt].l==Tree[rt].r)
    {
        Tree[rt].ms-=x;
        return ;
    }
    if(x<=Tree[rt<<1].ms)
        Update(x,rt<<1);
    else
        Update(x,rt<<1|1);
    Push_Up(rt);
}
void query_ans(int x,int rt)
{
    if(Tree[rt].l==Tree[rt].r)
    {
        num = Tree[rt].l;
        return ;
    }
    if(Tree[rt<<1].ms>=x)
        query_ans(x,rt<<1);
    else
        query_ans(x,rt<<1|1);
    Push_Up(rt);
}
int main()
{
    while(~scanf("%d%d%d",&h,&w,&n))
    {
        if(h>n) h =n;
        Build(1,h,1);
        while(n--)
        {
            int x;
            scanf("%d",&x);
            if(Tree[1].ms<x)puts("-1");
            else
            {
                num = -1;
                query_ans(x,1);
                printf("%d
",num); Update(x,1); } } } return 0; }