HDU2795线段树之单点更新

<a target=_blank href="http://acm.hdu.edu.cn/showproblem.php?pid=2795">点击打开链接</a>

题意：在一块h*w的矩形板上，要求贴1*Wi的广告，尽量往左往上的贴广告，如果贴不上的话，输出-1；否则输出这个小广告贴在了第几行上。

分析：可以用线段树保存每一行的长度，贴上广告之后便更新自己的值，线段树可以做到快速的查询和更新，下面有注释。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=200010;
const int inf=0x3f3f3f3f;
int num[maxn*4],h,w,n;
//线段树无改动模板
void buildtree(int le,int ri,int node){
    num[node]=w;
    if(le==ri) return ;
    int t=(le+ri)>>1;
    buildtree(le,t,node<<1);
    buildtree(t+1,ri,node<<1|1);
}
int query(int add,int le,int ri,int node){
    if(le==ri){
        num[node]-=add;//在查询时直接更新
        return le;//返回满足条件的行
    }
    int t=(le+ri)>>1;
    int ans;
    if(num[node<<1]>=add) ans=query(add,le,t,node<<1);
    else ans=query(add,t+1,ri,node<<1|1);
    num[node]=max(num[node<<1],num[node<<1|1]);
    return ans;
}
//线段树结束
int main(){
    int a;
    while(scanf("%d%d%d",&h,&w,&n)!=-1){
        if(h>n) h=n;//最多n个木棍，多的没有用，降低下复杂度
        buildtree(1,h,1);
        while(n--){
            scanf("%d",&a);
            if(num[1]<a) printf("-1\n");//num数组保存的是最大值，如果所有行的最大值小于a，自然输出-1；
            else printf("%d\n",query(a,1,h,1));
        }
    }
    return 0;
}