HDU2795 Billboard 线段树

开始的代码，暴力求解。。。。。果断超时

#include <iostream>
#include <cstdio>
#include <string.h>
using namespace std;

const int maxn = 200005;

int main()
{
    int  h, w, k, b;
    int  a[maxn] ;
    bool flag;
    while(scanf("%d%d%d", &h, &w, &k) != EOF){
        memset(a, 0, sizeof(a));
        for(int j = 0; j < k; j++){
            scanf("%d", &b);
            flag = false;
            for(int i = 1; i <= h; i++){
                if(a[i] + b <= w) {
                    a[i] += b;
                    flag = true;
                    printf("%d\n", i);
                    break;
                }
            }
            if(!flag) printf("-1\n");
        }
    }
    return 0;
}


线段树求解  

#include<iostream>
#include<cstdio>
using namespace std;
const int  N = 200005;
int h,w,n;
struct  Tree
{
    int l,r;
    int num;
}tree[N<<2];
int minn(int h,int n)
{
    if(h>n)
        return n;
    else
        return h;
}
int maxx(int a,int b)
{
    if(a>b)
        return a;
    else
        return b;
}
void creat(int i,int l,int r)
{
    int mid=(l+r)>>1;
    tree[i].l=l;
    tree[i].r=r;
    tree[i].num=w;
    if(l==r)
        return;
    creat(i<<1,l,mid);
    creat(i<<1|1,mid+1,r);
}
int query(int i,int len)
{
    if(tree[i].l==tree[i].r)
    {
        tree[i].num-=len;
        return tree[i].l;
    }
    else
    {
        int sum1=0,sum2=0;
        if(len<=tree[i<<1].num)
            sum1=query(i<<1, len);
        else
            if(len<=tree[i<<1|1].num)
                sum2=query(i<<1|1,len);
        tree[i].num=maxx(tree[i<<1].num,tree[i<<1|1].num);   //回溯
        return sum1+sum2;
    }
}
int main()
{
    while(scanf("%d%d%d",&h,&w,&n) != EOF)
    {
        int len;
        creat(1,1,minn(h,n));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&len);
            if(tree[1].num>=len)
                printf("%d\n",query(1,len));
            else
                printf("-1\n");
        }
    }
    return 0;
}