解析:这个是一个挺有意思的线段树,我们将第一行看作线段树最左端节点L,最后一行看作线段树最右节点R,宽度作为权值w,然后根据题目要求,每次先遍历左子树,若左子树装不下广告了,再遍历右子树。
代码如下:
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn = 200000 + 10;
int w, h, n;
struct Node
{
int w;
}tree[4 * maxn];
void build(int L, int R, int k)
{
if (L == R) {
tree[k].w = w;//构造最大值线段树
return;
}
int mid = (L + R) / 2;
build(L, mid, 2 * k);
build(mid + 1, R, 2 * k + 1);
tree[k].w = max(tree[2 * k].w, tree[2 * k + 1].w);
}
int update(int p, int L,int R,int k)
{
if (L == R) {
tree[k].w -= p;
return L;
}
int mid = (L + R) / 2;
int ans;
if (tree[2 * k].w >= p)//如果左子树还有剩余值,就查找左子树(因为左子树行数小)
ans = update(p, L, mid, 2 * k);
else
ans = update(p, mid + 1, R, 2 * k + 1);
tree[k].w = max(tree[2 * k].w, tree[2 * k + 1].w);
return ans;
}
int main()
{
ios::sync_with_stdio(false);
while (cin >> h >> w >> n)
{
if (h > n)h = n;//不加这个会RE,因为h会开到是1e9
build(1, h, 1);
int W;
for (int i = 1; i <= n; i++) {
cin >> W;
if (tree[1].w < W)
cout << -1 << endl;
else
cout << update(W, 1, h, 1) << endl;
}
}
return 0;
}