输入三个整数,h, w, n。代表 h×w 的公告板和接下来有n个公告。
接下来输入n个公告的宽度Ki。高度都为1。
公告的规则是尽量往最上面放置。
问这n个公告分别放置的位置是第几行。
如果不能放置,输出-1。第一眼看h,w比较大,109, 但是n的最大值是200000。
当h > n的时候,超出的部分是一定不会有公告放置的。
所以就可以用线段树来做了。
构造[1, min(n, h)]的线段树,每个叶子结点代表公告的一行。
线段树的每个结点存储的是左右子树的最大值。
然后不断更新即可。
#include <stdio.h>
#include <algorithm>
#define lson rt << 1
#define rson rt << 1 | 1
using namespace std;
int maxn[200005 << 2];
int h, w, n;
void build(int l, int r, int rt)
{
maxn[rt] = w;
if(l == r)
{
return ;
}
int mid = (l + r) >> 1;
build(l, mid, lson);
build(mid + 1, r, rson);
}
void update(int l, int r, int rt, int val, int where)
{
if(l == r)
{
maxn[rt] -= val;
return ;
}
int mid = (l + r) >> 1;
if(mid >= where) update(l, mid, lson, val, where);
else update(mid + 1, r, rson, val, where);
maxn[rt] = max(maxn[lson], maxn[rson]);
}
int query(int l, int r, int rt, int val)
{
if(l == r && maxn[rt] >= val)
{
return l;
}
int mid = (l + r) >> 1;
if(maxn[lson] >= val) return query(l, mid, lson, val);
else if(maxn[rson] >= val) return query(mid + 1, r, rson, val);
return -1;
}
int main()
{
while(~scanf("%d %d %d", &h, &w, &n))
{
int k = min(n, h);
build(1, k, 1);
for(int i = 0; i < n; i++)
{
int temp;
scanf("%d", &temp);
int t = query(1, k, 1, temp);
printf("%d\n", t);
if(t != -1) update(1, k, 1, temp, t);
}
}
return 0;
}