-
Notifications
You must be signed in to change notification settings - Fork 13
/
Copy path218.天际线问题.py
66 lines (60 loc) · 2.66 KB
/
218.天际线问题.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# 城市的天际线是从远处观看该城市中所有建筑物形成的轮廓的外部轮廓。现在,假设您获得了城市风光照片(图A)上显示的所有建筑物的位置和高度,请编写一个程序以输出由这些建筑物形成的天际线(图B)。
#
#
#
# 每个建筑物的几何信息用三元组 [Li,Ri,Hi] 表示,其中
# Li
# 和
# Ri
# 分别是第
# i
# 座建筑物左右边缘的
# x
# 坐标,Hi
# 是其高度。可以保证 0 ≤ Li, Ri ≤ INT_MAX, 0 < Hi ≤ INT_MAX
# 和
# Ri - Li > 0。您可以假设所有建筑物都是在绝对平坦且高度为
# 0
# 的表面上的完美矩形。
#
# 例如,图A中所有建筑物的尺寸记录为:[[2 9 10], [3 7 15], [5 12 12], [15 20 10], [19 24 8]] 。
#
# 输出是以 [[x1, y1], [x2, y2], [x3, y3], ...]
# 格式的“关键点”(图B中的红点)的列表,它们唯一地定义了天际线。关键点是水平线段的左端点。请注意,最右侧建筑物的最后一个关键点仅用于标记天际线的终点,并始终为零高度。此外,任何两个相邻建筑物之间的地面都应被视为天际线轮廓的一部分。
#
# 例如,图B中的天际线应该表示为:[[2 10], [3 15], [7 12], [12 0], [15 10], [20 8], [24, 0]]。
#
# 说明:
#
# 任何输入列表中的建筑物数量保证在[0, 10000] 范围内。
# 输入列表已经按左 x
# 坐标 Li 进行升序排列。
# 输出列表必须按
# x
# 位排序。
# 输出天际线中不得有连续的相同高度的水平线。例如[...[2
# 3], [4 5], [7 5], [11 5], [12 7]...] 是不正确的答案;三条高度为
# 5
# 的线应该在最终输出中合并为一个:[...[2 3], [4 5], [12 7], ...]
import heapq
class Solution:
def getSkyline(self, buildings: List[List[int]]) -> List[List[int]]:
if not buildings: return []
points = []
heap = [[0, float('inf')]]
res = [[0, 0]]
# 将所有断电加入到点集中(每个建筑物的左右端点)
for l, r, h in buildings:
points.append((l, -h, r)) # 负号将最小堆,变成了最大堆
points.append((r, h, 0)) # r的右端点为0
# 将端点从小到大排序
points.sort() # 如果当前点相等,则按照高度升序
# 遍历每个点,分别判断出堆,入堆,添加关键点操作
for l, h, r in points:
while l >= heap[0][1]:
heapq.heappop(heap)
if h < 0:
heapq.heappush(heap, [h, r])
if res[-1][1] != -heap[0][0]:
res.append([l, -heap[0][0]])
return res[1:]