HDU2795线段树入门 简单查询和修改

#include<iostream> using namespace std; const int n=1e6+20; int h,w,x; struct node
{ int r; int l; int res;
};
node tree[n*4]; void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].res=w; if(l==r)
    { return ;
    } int mid=(l+r)>>1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
} int query(int root,int key)
{ if(tree[root].l==tree[root].r)
    {
        tree[root].res-=key; return tree[root].r;
    } int c; if(tree[root<<1].res>=key) c=query(root<<1,key); else c=query(root<<1|1,key);
    tree[root].res=max(tree[root<<1].res,tree[root<<1|1].res); // cout<<c<<endl; return c;
} int main(){ while(~scanf("%d%d%d",&h,&w,&x)){ if(h>x) h=x;
        build(1,1,h); while(x--)
        { int value;
            scanf("%d",&value); if(tree[1].res<value) printf("-1\n"); else printf("%d\n",query(1,value));
        }
    }
}