题意:一个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;
}