计蒜客公告板

题目

蒜厂有一个 h×w 的矩形公告板，其中 h 是高度，w 是宽度。
现在有若干张 1×Wi 的公告， Wi 是宽度，公告只能横着放，即高度为 1 的边垂直于水平面，且不能互相有重叠，每张公告都要求尽可能的放在最上面的合法的位置上。
若可以放置，输出每块可放置的位置的行号；若不存在，输出 −1。行号由上至下分别为 1,2,…,h。
输入格式
第一行三个整数 h，w，n (1≤h,w≤10^9;1≤n≤200,000) 。
接下来 n 行，每行一个整数 Wi(1≤Wi≤109) 。
输出格式
输出n 行，一行一个整数。

分析

核心思想：线段shu，节点s[p]表示其范围行的总剩余宽度之和，

AC代码(别用cin,cout，这是个坑)

#include<iostream>
#include<cstdio>
using namespace std;

const long MAXH=200001;
int s[4*MAXH];//s[p]（p表示层序遍历中树状节点的下标）表示其区间行的所有剩余宽度之和
int h,w,n,flag=0;

void buildtree(int p,int l,int r){//建树
	if(l==r){//将每个叶子节点都赋为最大宽度w
        s[p]=w;
        return;
    }
    int mid=(l+r)/2;
    buildtree(p*2,l,mid);//建立左子树
    buildtree(p*2+1,mid+1,r);//建立右子树
    s[p]=s[p*2]+s[p*2+1];//更新父亲信息
    return;
}
void modify(int p,int l,int r,int v){//填数并更新线段树
    if(l==r){//找到可以放入公共板的行
        s[p]-=v;
        flag=1;
        printf("%d\n",l);
        return;
    }
    int mid=(l+r)/2;
    if(s[p*2]>=v){//左儿子的宽度够
        modify(p*2,l,mid,v);
    }
    if(s[p*2+1]>=v&&!flag){//右儿子宽度够，且这个数还没有放入行中
        modify(p*2+1,mid+1,r,v);
    }
    s[p]=s[p*2]+s[p*2+1];
    return;
}

int main()
{
    cin>>h>>w>>n;
    buildtree(1,1,h);
    int num;
    while(n--){
        scanf("%d",&num);
        flag=0;
        if(s[1]>=num)
        	modify(1,1,h,num);
        if(!flag)
            printf("-1\n");
    }
    return 0;
}