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

 

题目传送门

 1 /*  2  主要利用线段树求区间最值，sum[]代表位置可用空间  3  每次找到最大值的位置  4  功能：查询最靠前能容纳广告的位置  5 */  6 #include <cstdio>  7 #include <cstring>  8 #include <algorithm>  9 #define lson l, m, rt << 1 10 #define rson m+1, r, rt << 1 | 1 11 12 const int MAX_N = 2000000 + 10; 13 int w[MAX_N]; 14 int num[MAX_N]; 15 int sum[MAX_N << 2]; 16 int n, h, W; 17 18 void maxsum(int rt) 19 { 20 sum[rt] = std::max (sum[rt << 1], sum[rt << 1 | 1]); 21 } 22 23 void build(int l, int r, int rt) 24 { 25 sum[rt] = W; 26 if (l == r) return ; 27 int m = (l + r) >> 1; 28  build (lson); 29  build (rson); 30 } 31 32 int query(int p, int l, int r, int rt) 33 { 34 if (l == r) 35  { 36 sum[rt] -= p; //如果可以用，更新sum，update在query里做了 37 return l; //返回该位置 38  } 39 int m = (l + r) >> 1; 40 int ans; 41 if (p <= sum[rt << 1]) //如果p小于左孩子的最大值，则往左边下去 42 ans = query (p, lson); 43 else 44 ans = query (p, rson); 45  maxsum (rt); 46 47 return ans; 48 } 49 50 int main(void) //HDOJ 2795 Billboard 51 { 52 //freopen ("inD.txt", "r", stdin); 53 54 while (~scanf ("%d%d%d", &h, &W, &n)) 55  { 56 if (h > n) // h <= n 57 h = n; 58 59 build (1, h, 1); 60 61 int x; 62 while (n--) 63  { 64 scanf ("%d", &x); 65 if (x > sum[1]) printf ("%d\n", -1); //所有位置的最大值 66 else printf ("%d\n", query(x, 1, h, 1)); 67  } 68  } 69 70 return 0; 71 }