线段树(单点更新) HDOJ 2795 Billboard

题目传送门

 /*
     主要利用线段树求区间最值，sum[]代表位置可用空间
     每次找到最大值的位置
     功能：查询最靠前能容纳广告的位置
 */
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define lson l, m, rt << 1
 #define rson m+1, r, rt << 1 | 1

 const int MAX_N =  + ;
 int w[MAX_N];
 int num[MAX_N];
 int sum[MAX_N << ];
 int n, h, W;

 void maxsum(int rt)
 {
     sum[rt] = std::max (sum[rt << ], sum[rt <<  | ]);
 }

 void build(int l, int r, int rt)
 {
     sum[rt] = W;
     if (l == r) return ;
     int m = (l + r) >> ;
     build (lson);
     build (rson);
 }

 int query(int p, int l, int r, int rt)
 {
     if (l == r)
     {
         sum[rt] -= p;        //如果可以用，更新sum，update在query里做了
         return l;            //返回该位置
     }
     int m = (l + r) >> ;
     int ans;
     if (p <= sum[rt << ])        //如果p小于左孩子的最大值，则往左边下去
         ans = query (p, lson);
     else
         ans = query (p, rson);
     maxsum (rt);

     return ans;
 }

 int main(void)        //HDOJ 2795 Billboard
 {
     //freopen ("inD.txt", "r", stdin);

     while (~scanf ("%d%d%d", &h, &W, &n))
     {
         if (h > n)        // h <= n
             h = n;

         build (, h, );

         int x;
         while (n--)
         {
             scanf ("%d", &x);
             if (x > sum[])    printf ("%d\n", -);    //所有位置的最大值
             else    printf ("%d\n", query(x, , h, ));
         }
     }

     return ;
 }