用Python找出视野更好的建筑物的程序
假设,我们提供了一个包含不同建筑物高度的数组。建筑物位于一条线上,如果建筑物不被另一座更高的建筑物阻挡,它的视野会更好。因此,如果提供包含高度的数组,我们必须找出没有其他更高建筑物来阻挡视线的建筑物。索引从满足条件的数组中返回。
因此,如果输入类似于height=[5,6,8,7],那么输出将是[2,3]。数组索引0和1中的建筑物被索引2处的建筑物挡住。索引2和3处的建筑物没有被阻挡,因为位置2处的较高建筑物位于位置3处的较短建筑物后面。
示例
让我们看看以下实现以获得更好的理解-
def solve(heights): res, h = [], 0 for i in range(len(heights) - 1, -1, -1): if heights[i] > h: res.append(i) h = heights[i] return res[::-1] print(solve([5, 6, 8, 7]))
输入
[5, 6, 8, 7]输出结果
[2, 3]