题目中给出的h和w范围均大,其实n的最大范围才200000,所以我们建立的线段树大小为min(h,n),线段树的每一个节点包含一个变量c,记录当前区间内还剩下的可以put on的最大长度。插入一个数时,如果该数大于该区间最大值,则返回-1,说明put on不了。否则将它插入到页节点,并返回插入的下标,接着一定不要忘记更新父节点的c值。
#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN = 200001;
struct Tnode{
int e, b;
int c;
};
int h, w, n,ans;;
Tnode tree[4 * MAXN];
void Create(int v, int b, int e){
tree[v].b = b;
tree[v].e = e;
tree[v].c = w;
if (e > b){
int mid = (b + e) >> 1;
Create(2 * v + 1, b, mid);
Create(2 * v + 2, mid + 1, e);
}
}
void Insert(int v,int x){
if (x > tree[v].c){
ans = -1;
return;
}
if (tree[v].b==tree[v].e){
tree[v].c -= x;
ans = tree[v].b;
return;
}
if (x <= tree[2 * v + 1].c)
Insert(2 * v + 1, x);
else
Insert(2 * v + 2, x);
tree[v].c = max(tree[2 * v + 1].c, tree[2 * v + 2].c);
}
int main(){
int x;
while (~scanf("%d%d%d", &h, &w, &n)){
int len = min(h, n);
Create(0, 1, len);
for (int i = 0; i < n; i++){
scanf("%d",&x);
Insert(0, x);
printf("%d\n", ans);
}
}
return 0;
}
原文:http://www.cnblogs.com/td15980891505/p/5747735.html