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

http://acm.hdu.edu.cn/showproblem.php?pid=2795

#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));
        }
    }
}