线段树专辑之单点更新4
题意略。
在做题前我首先注意到了数据的范围。w和h均为10^9,如果开4倍这么大的数据,必超无疑。所幸,大牛给的例题没有用到离散化。根据题目的意思我们知道最多只有n个广告,而每个广告注定只有1单位长。而n的范围只有200,000. 撑死也就200000行,一开始先比较一下h和n的值,之后选取较小的进行建树即可。
线段树每个节点的key值储存区间内的所有子节点中的最大宽度。如果最大宽度大于广告长度,则往下搜!
我用了一个vis表示是否已经搜到符合要求的子节点的位置。如果搜到了则标记为true表示以搜到。注意要用引用或者指针作为参数传值。这样才能保证vis的实时更新。
(如果不用vis标记的话每次都要搜遍所有的节点!等同于暴力!)
#include <iostream>
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
const int maxx = 3*1e5;
struct po
{
int lson,rson;
long long key;
}seg[maxx<<2];
void build(int left, int right, int dex,LL w)
{
seg[dex].lson = left;
seg[dex].rson = right;
if(left == right)
{
seg[dex].key = w;
return ;
}
int mid = (left+right)/2;
build(left,mid,dex<<1,w);
build(mid+1,right,dex<<1|1,w);
seg[dex].key = w;
}
LL query(int x, int dex,bool& vis)
{
if(seg[dex].lson == seg[dex].rson)
{
if(seg[dex].key>=x && !vis)
{
seg[dex].key-=x;
vis = true;
return seg[dex].lson;
}
else
return 0;
}
// int mid = (seg[dex].lson + seg[dex].rson)/2;
if(seg[dex].key>=x && !vis)
{
LL tem = query(x,dex<<1,vis) + query(x,dex<<1|1,vis);
seg[dex].key = max(seg[dex<<1].key,seg[dex<<1|1].key);
return tem;
}
else
return 0;
}
int main()
{
// freopen("2795.txt","r",stdin);
LL h,w,n;
while(~scanf("%I64d%I64d%I64d",&h,&w,&n))
{
n>=h?build(1,h,1,w):build(1,n,1,w);
bool vis;
for(int i=1;i<=n;i++)
{
vis = false;
LL tem;
scanf("%I64d",&tem);
LL ans = query(tem,1,vis);
if(!ans) printf("-1\n");
else printf("%I64d\n",ans);
}
}
return 0;
}