POJ 3667 Hotel【经典的线段树】
POJ 3667 Hotel
http://poj.org/problem?id=3667
题目大意：
奶牛们最近的旅游计划，是到苏必利尔湖畔，享受那里的湖光山色，以及
明媚的阳光。作为整个旅游的策划者和负责人，贝茜选择在湖边的一家著名的
旅馆住宿。这个巨大的旅馆一共有N (1 <= N <= 50,000)间客房，它们在同一层
楼中顺次一字排开，在任何一个房间里，只需要拉开窗帘，就能见到波光粼粼的
湖面。
贝茜一行，以及其他慕名而来的旅游者，都是一批批地来到旅馆的服务台，
希望能订到D_i (1 <= D_i <= N)间连续的房间。服务台的接待工作也很简单：
如果存在r满足编号为r..r+D_i-1的房间均空着，他就将这一批顾客安排到这些
房间入住；如果没有满足条件的r，他会道歉说没有足够的空房间，请顾客们另
找一家宾馆。如果有多个满足条件的r，服务员会选择其中最小的一个。
旅馆中的退房服务也是批量进行的。每一个退房请求由2个数字X_i、D_i
描述，表示编号为X_i..X_i+D_i-1 (1 <= X_i <= N-D_i+1)房间中的客人全部
离开。退房前，请求退掉的房间中的一些，甚至是所有，可能本来就无人入住。
而你的工作，就是写一个程序，帮服务员为旅客安排房间。你的程序一共
需要处理M (1 <= M < 50,000)个按输入次序到来的住店或退房的请求。第一个
请求到来前，旅店中所有房间都是空闲的。
分析：
线段树，节点内存入左连续块，右连续块，以及当前区间内的最大连续块
将订房服务分为两步：
1.查询是否存在d连续块，若存在，返回r
2.根据1，进行实际的住房操作，即将[r,r+d-1]区间住满
退房服务，与2类似
View Code
#include
<
stdio.h
>
const
int
N
=
50000
+
10
;
/*
节点
*/
struct
node
{
int
left,right;
int
cval;
//
该区间内最大的连续块长度
int
lval,rval;
//
左、右边界连续块得长度
};
node tree[N
*
3
];
//
将id所对应的区间置满或置空
inline
void
initNode(
int
id,
int
state)
{
int
len
=
tree[id].right
-
tree[id].left
+
1
;
tree[id].cval
=
state
*
len;
tree[id].lval
=
tree[id].cval;
tree[id].rval
=
tree[id].cval;
}
inline
int
max(
int
a,
int
b)
{
return
a
>
b
?
a:b;
}
//
获得id所在区间的长度
inline
int
getLen(
int
id)
{
return
tree[id].right
-
tree[id].left
+
1
;
}
//
建树
void
createTree(
int
left,
int
right,
int
id)
{
tree[id].left
=
left;
tree[id].right
=
right;
initNode(id,
1
);
if
(tree[id].left
==
tree[id].right)
return
;
int
mid
=
(left
+
right)
>>
1
;
createTree(left,mid,
2
*
id);
createTree(mid
+
1
,right,
2
*
id
+
1
);
}
//
查询是否存在长度为need的连续段，若存在返回其左端所在位置
int
search(
int
need,
int
id)
{
if
(tree[id].cval
<
need)
return
0
;
if
(tree[id].left
==
tree[id].right)
return
tree[id].left;
if
(tree[id].lval
>=
need)
return
tree[id].left;
if
(tree[
2
*
id].cval
>=
need)
return
search(need,
2
*
id);
else
if
(tree[
2
*
id].rval
+
tree[
2
*
id
+
1
].lval
>=
need)
return
tree[
2
*
id].right
-
tree[
2
*
id].rval
+
1
;
else
return
search(need,
2
*
id
+
1
);
}
/*
更新left,right区间，将其置空或置满
*/
void
update(
int
left,
int
right,
int
id,
int
state)
{
int
len
=
tree[id].right
-
tree[id].left
+
1
;
if
(tree[id].left
>=
left
&&
tree[id].right
<=
right)
{
initNode(id,state);
return
;
}
int
mid
=
(tree[id].left
+
tree[id].right)
>>
1
;
if
(tree[id].cval
==
len
||
tree[id].cval
==
0
)
{
int
st
=
0
;
if
(tree[id].cval
!=
0
)st
=
1
;
initNode(
2
*
id,st);
initNode(
2
*
id
+
1
,st);
}
if
(right
<=
mid)
update(left,right,
2
*
id,state);
else
if
(left
>
mid)
update(left,right,
2
*
id
+
1
,state);
else
{
update(left,mid,
2
*
id,state);
update(mid
+
1
,right,
2
*
id
+
1
,state);
}
tree[id].lval
=
tree[
2
*
id].lval;
if
(tree[id].lval
==
getLen(
2
*
id))
tree[id].lval
+=
tree[
2
*
id
+
1
].lval;
tree[id].rval
=
tree[
2
*
id
+
1
].rval;
if
(tree[id].rval
==
getLen(
2
*
id
+
1
))
tree[id].rval
+=
tree[
2
*
id].rval;
tree[id].cval
=
tree[
2
*
id].rval
+
tree[
2
*
id
+
1
].lval;
tree[id].cval
=
max(tree[
2
*
id].cval,tree[id].cval);
tree[id].cval
=
max(tree[
2
*
id
+
1
].cval,tree[id].cval);
}
int
main()
{
int
n,m;
scanf(
"
%d%d
"
,
&
n,
&
m);
createTree(
1
,n,
1
);
int
f,d,x;
while
(m
--
)
{
scanf(
"
%d
"
,
&
f);
if
(f
==
1
)
{
scanf(
"
%d
"
,
&
d);
int
r
=
search(d,
1
);
printf(
"
%d\n
"
,r);
if
(r)
{
update(r,r
+
d
-
1
,
1
,
0
);
//
放满
}
}
else
{
scanf(
"
%d%d
"
,
&
x,
&
d);
update(x,x
+
d
-
1
,
1
,
1
);
}
}
return
0
;
}