搞了两天的线段树,我一直在犯一个错误,代码看似没错提交总是RE!,心中的迷惑不解,连续做了几道题,都这样,我彻底无语了,搞的我没心情去涮题了。
这题也不列外,提交总是RE,感觉自己数组开的完全没问题,在网上搜了很久,以为和我写得差不多的代码竟可以AC!为何?,我试着把他的代码改一改,就在我改了全局变量mid后时候。
提交终于出问题了。心中迷惑豁然已解!,原来是全局变量mid若的祸!,但我还是不解,为什么不能这样!发现在query函数里有多个嵌套。应该是mid在传递过程中,由于定义是全局变量,(我们知道,全局变量只需要定义一次就行了)所以在以后的循环中没有再次定义,导致mid的值是以前的值,这样嵌套循环出现了严重的错误!。
改了这个错误,数组没开错的话,提交就AC了!
说一下解题思路:每次寻找最大的位置,用线段树去维护,能够粘贴的地方尽量靠左,也就是说:如果左边的位置能够容的下,就往左边找。找到的位置减去长度L。
代码如下:
#include<cstdio>
#include<algorithm>
using namespace std;
int num[900000];
int w;
void build(int l,int r,int i){
num[i]=w;
if(l==r) return;
int mid=(l+r)>>1;
build(l,mid,i<<1);
build(mid+1,r,i<<1|1);
}
inline int query(int a,int l,int r,int i){
if(l==r){
num[i]-=a;
return l;
}
int mid=(l+r)>>1;
int ret=(num[i<<1]>=a) ? query(a,l,mid,i<<1):query(a,mid+1,r,i<<1|1);
num[i]=max(num[i<<1],num[i<<1|1]);
return ret;
}
int main(){
int h,n,x;
while(~scanf("%d %d %d",&h,&w,&n)){
if(h>n) h=n;
build(1,h,1);
while(n--){
scanf("%d",&x);
if(x>num[1]) puts("-1");
else printf("%d\n",query(x,1,h,1));
}
}
return 0;
}