在 Python 中从可以看到大海的地方查找建筑物的程序
假设我们有一个不同建筑物的高度列表。具有高度值heights[i]的建筑物可以在其右侧的每个建筑物都比该建筑物短时看到海洋。我们必须按升序找到可以看到海洋的建筑物索引。
因此,如果输入类似于heights=[8,12,12,9,10,6],那么输出将是[2,4,5]因为我们可以在索引2处从建筑物高度12看到海洋,从索引10处的建筑高度10和索引5处的最后一座建筑。
示例
让我们看看以下实现以获得更好的理解-
def solve(heights): stack = [] for idx, h in enumerate(heights): while stack and heights[stack[-1]] <= h: stack.pop() stack.append(idx) return stack heights = [8, 12, 12, 9, 10, 6] print(solve(heights))
输入
[8, 12, 12, 9, 10, 6]输出结果
[2, 4, 5]