Leetcode 1762. Buildings With an Ocean View [Python]

Case_CaiNiao

于 2021-12-10 17:26:40 发布

阅读量853

点赞数

分类专栏： Leetcode学习记录文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/sinat_30403031/article/details/121861357

版权

Leetcode学习记录专栏收录该内容

268 篇文章

订阅专栏

该篇博客介绍了一个Python算法，用于找到一组高度中最高建筑物的索引。算法通过维护一个堆栈，确保堆栈顶部的建筑物高度始终大于或等于即将入栈的建筑物。如果新建筑物高度更大，则直接入栈；否则，弹出高度较小的堆栈顶部元素，直到找到合适的位置。最终堆栈中的元素即为最高建筑物的索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

考查stack。stack顶的元素一定要比待入stack的元素大，否则弹出stack顶元素。

class Solution:
    def findBuildings(self, heights: List[int]) -> List[int]:
        stack = []
        for index, B in enumerate(heights):
            if not stack or stack[-1][1] > B:
                stack.append((index, B))
            else: 
                while stack and stack[-1][1] <= B:
                    stack.pop(-1)
                stack.append((index, B))
        res = []
        for index, B in stack:
            res.append(index)
        return res