poj3667【线段树】/【类似权值线段树写法】

题意：n个空房间。两种操作：1.选择最小的连续D个房间入住，并输出这连续D个房间的最小标号。2.将某个区间内的房间全部退房。

 #include <cstdio>
 #include <iostream>
 #include <algorithm>
 #define ll long long
 #define lson l, m, rt<<1
 #define rson m+1, r, rt<<1|1
 #define st first
 #define nd second
 #define mp make_pair
 #define pii pair<int, int>
 #define gg puts("gg");
 using namespace std;
 void gmax(int& a, int b){
     if(a < b) a = b;
 }
 const int N = 5e4+;
 struct Node{
     int lsum, rsum, sum;
     int tag;
 };
 Node T[N<<];
 void pushup(int l, int r, int rt){
     T[rt].lsum = T[rt<<].lsum, T[rt].rsum = T[rt<<|].rsum;
     int m = l+r >> ;
     if(T[rt<<].lsum == m-l+) T[rt].lsum += T[rt<<|].lsum;
     if(T[rt<<|].rsum == r-m) T[rt].rsum += T[rt<<].rsum;
     T[rt].sum = max(T[rt<<].sum, T[rt<<|].sum);
     gmax(T[rt].sum, T[rt<<].rsum+T[rt<<|].lsum);
 }
 void pushdown(int l, int r, int rt){
     if(T[rt].tag != -){
         T[rt<<].tag = T[rt<<|].tag = T[rt].tag;
         int m = l+r >> ;
         T[rt<<].lsum = T[rt<<].rsum = T[rt<<].sum = T[rt].tag? m-l+ : ;
         T[rt<<|].lsum = T[rt<<|].rsum = T[rt<<|].sum = T[rt].tag? r-m : ;
         T[rt].tag = -;
     }
 }
 void build(int l, int r, int rt){
     T[rt].lsum = T[rt].rsum = T[rt].sum = r-l+;
     T[rt].tag = -;
     if(l == r)
         return ;
     int m = l+r >> ;
     build(lson);
     build(rson);
 }
 int query(int c, int l, int r, int rt){
     //printf("query %d: l %d, r %d, lson %d, rson %d, sum %d\n", rt, l, r, T[rt].lsum, T[rt].rsum, T[rt].sum);
     if(l == r)
         return l;
     pushdown(l, r, rt);
     int m = l+r >> ;
     if(T[rt<<].sum >= c) return query(c, lson);
     if(T[rt<<].rsum+T[rt<<|].lsum >= c) return m-T[rt<<].rsum+;
     return query(c, rson);
 }
 void update(int L, int R, int c, int l, int r, int rt){
     if(L <= l&&r <= R){
         T[rt].tag = c;
         T[rt].lsum = T[rt].rsum = T[rt].sum = c? r-l+:;
         return ;
     }
     pushdown(l, r, rt);
     int m = l+r >> ;
     if(L <= m) update(L, R, c, lson);
     if(R > m) update(L, R, c, rson);
     pushup(l, r, rt);
     //printf("updaet %d: l %d, r %d, lson %d, rson %d, sum %d\n", rt, l, r, T[rt].lsum, T[rt].rsum, T[rt].sum);
 }

 int main(){
     int n, m, x, y, op;
     scanf("%d%d", &n, &m);
     build(, n, );
     while(m--){
         scanf("%d", &op);
         if(op == ){
             scanf("%d", &x);
             if(T[].sum < x) puts("");
             else {
                 int ret = query(x, , n, );
                 printf("%d\n", ret);
                 update(ret, ret+x-, , , n, );
             }
         }
         else {
             scanf("%d%d", &x, &y);
             update(x, x+y-, , , n, );
         }
     }
     return ;
 }

后记：这也是线段树一经典题。不难。

主要是通过这种写法可以O(logn)的时间内完成离散化查询。不过平时一般都是二分+树状数组O(lognlogn)完成离散化查询。