【转】poj 1823 ：Hotel (线段树)

题意：一个hotel，有n间连续的房间，现在有m组操作：

      type 1: '1, a, b': 第a个房间起的b个房间有旅客入住。
      type 2: '2, a, b': 第a个房间起的b个房间的旅客离开。
      type 3: '3': 问最长的连续空房间有多少间。

 

思路：线段树。这道题很好的利用线段树递归的性质，加深了对线段树递归的理解。复习了一下延迟的操作，学会了与延迟操作相反的操作，即利用递归，在递归回来的时候，由于左右子结点性质的改变，即时对父结点信息进行相应的更改，这个要注意。

 

源代码：(944K 3282MS)

#include<iostream>
using namespace std;
const int Max = 16005;

 

struct{
    int l, r;
    int lma, ma, rma;   // lma为这个区间左边的最长连续空房间的数量。
    int cover;          //‘1’表示这个区间的房间全住人，‘-1’表示全空，‘0’表示有空用住。
}node[3*Max];

 

int max(int a, int b){
    return a > b ? a : b;
}

 

void BuildTree(int left, int right, int u){       //  建树。
    node[u].l = left;
    node[u].r = right;
    if(left == right) return;
    int mid = (left + right)>>1;
    BuildTree(left, mid, u<<1);
    BuildTree(mid+1, right, (u<<1)+1);
}

 

void getdown(int u, int op){      //  延迟的操作。
    node[u].cover = 0;
    node[u<<1].cover = op;
    node[(u<<1)+1].cover = op;
    if(op == 1){
        node[u<<1].lma = 0;
        node[u<<1].ma = 0;
        node[u<<1].rma = 0;
        node[(u<<1)+1].lma = 0;
        node[(u<<1)+1].ma = 0;
        node[(u<<1)+1].rma = 0;
    }
    else{
        int len;
        len = node[u<<1].r - node[u<<1].l + 1;   
        node[u<<1].lma = len;
        node[u<<1].ma = len;
        node[u<<1].rma = len;
        len = node[(u<<1)+1].r - node[(u<<1)+1].l + 1;   
        node[(u<<1)+1].lma = len;
        node[(u<<1)+1].ma = len;
        node[(u<<1)+1].rma = len;
    }
}

 

void updata(int left, int right, int op, int u){   //  修改。
    if(left <= node[u].l && right >= node[u].r){
        node[u].cover = op;
        if(op == 1)
            node[u].lma = node[u].ma = node[u].rma = 0;
        else{   
            int len = node[u].r - node[u].l + 1;
            node[u].lma = node[u].ma = node[u].rma = len;
        }
        return;
    }
    if(node[u].cover == op) return;
    if(node[u].cover == -op) getdown(u, -op);
    if(right <= node[u<<1].r)
        updata(left, right, op, u<<1);
    else if(left >= node[(u<<1)+1].l)
        updata(left, right, op, (u<<1)+1);
    else{
        updata(left, right, op, u<<1);
        updata(left, right, op, (u<<1)+1);
    }

 

    //  *很好运用递归，由左右子结点的信息，对父结点的三个连续区间最大值进行相应的更改。

    if(node[u<<1].cover == -1)                       //  求父结点的lma。
        node[u].lma = node[u<<1].ma + node[(u<<1)+1].lma;
    else
        node[u].lma = node[u<<1].lma;
    if(node[(u<<1)+1].cover == -1)                   //  求父结点的rma。
        node[u].rma = node[(u<<1)+1].ma + node[u<<1].rma;
    else
        node[u].rma = node[(u<<1)+1].rma;
    int a = node[u<<1].rma + node[(u<<1)+1].lma;     //  求父结点的ma。
    int b = max(node[u<<1].ma, node[(u<<1)+1].ma);
    int c = max(node[u].lma, node[u].rma);
    node[u].ma = max(max(a, b), c);

 

    //  *递归回来的时候，由于左右子结点性质的改变，必须对父结点信息进行相应的更改，WA在了这里。

    if(node[u<<1].cover == node[(u<<1)+1].cover)
        node[u].cover = node[u<<1].cover;
}

 

int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    BuildTree(1, n, 1);
    node[1].cover = -1;   
    node[1].ma = n;
    while(m --){
        int op, a, b;
        scanf("%d", &op);
        if(op == 1){
            scanf("%d%d", &a, &b);
            updata(a, a+b-1, 1, 1);    //  为a+b-1，不能算成a+b。
        }
        else if(op == 2){
            scanf("%d%d", &a, &b);
            updata(a, a+b-1, -1, 1);
        }
        else
            printf("%d\n", node[1].ma);
    }
    return 0;
}