题目就不重复了。还是相当难的一道题,看别人的解法看了N久……
基本思想还是从左到右遍历所有的节点,建立一个数据结构保存已经遍历过的、当下对最终结果有影响的那些元素。参考了参考答案中排名第一的解法,其巧妙之处有二,一是直接利用了python中的堆排序模块,就避免了手动排序的麻烦,二是先对左边界排序,确保遍历的顺序,再对高度排序,就可以取到最高值。
代码如下(非原创,只是自己看懂得很艰难,于是加了注释):
class Solution(object):
def getSkyline(self, buildings):
#第一步 把所有的矩阵化成线条的形式,用左-右-负高度的形式表示,第二项表示右边界也就是“零开始的线条”,用set是为了防止两个矩形在同一点结束,用tuple是因为set的元素不能是list
events = sorted([(L,-H,R) for L,R,H in buildings] + list({(R,0,None) for _,R,_ in buildings}))
#res是最终结果,hp表示“现在需要考虑的线段们”
res , hp = [[0,0]] , [(0,float("inf"))]
#对所有的线条进行遍历
for x, negH, R in events:
#当一个线条的左边界大于hp中线条的右边界时,说明这个线条已经没有任何作用了,可以删去
while x >= hp[0][1]:
heapq.heappop(hp)
#如果这是一个非零线段,我们把它放入hp中
if negH:
heapq.heappush(hp,(negH , R))
#这里需要注意,heapq保证列表中的第一个元素一定是最小元素,取相反数也就是目前最高元素。
#res[-1][1]是结果中最后一个元素的高度,hp[0][0]是目前最高那个元素的高度,如果不相等,就需要更新res。
#为什么我们只考虑dp中第一个元素?首先,这个元素在dp里面,所以它是有影响的,其次,它是最高的,当前我们只要最高的元素,至于比它低的元素,都被覆盖了
if res[-1][1] + hp[0][0]:
res.append([x,-hp[0][0]])
return res[1:]