leetcode63-天际线问题-python

题目就不重复了。还是相当难的一道题,看别人的解法看了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:]

猜你喜欢

转载自blog.csdn.net/weixin_39655021/article/details/84929018