LeetCode：962. Maximum Width Ramp - Python

2024年6月11日 165次阅读

962. 最大宽度坡

问题描述：

给定一个整数数组 A，坡是元组 (i, j)，其中 i < j且 A[i] <= A[j]。这样的坡的宽度为j - i。

找出 A 中的坡的最大宽度，如果不存在，返回 0。

示例 1：

输入：[6,0,8,2,1,5]
输出：4
解释：最大宽度的坡为 (i, j) = (1, 5): A[1] = 0 且 A[5] = 5.

示例 2：

输入：[9,8,1,0,1,9,4,0,4,1]
输出：7
解释：最大宽度的坡为 (i, j) = (2, 9): A[2] = 1 且 A[9] = 1.

提示：

  1. 2 <= A.length <= 50000
  2. 0 <= A[i] <= 50000

问题分析：

好无趣，只能刷题玩。看思路：

（1）因为，要求A[i] <= A[j]，所以可以先对列表从小到大排序，排序之后，就开始比较这些元素之前的位置，获取最优解即可。更简单一点就是，直接获取排序好的下标索引。

（2）然后依次判断下标索引，获得最优值。

        # 获取列表A排序好的下标索引，功能类似于 numpy.argsort，下面这两行代码等价
        sorted(range(len(A)), key=lambda i:A[i])
        sorted(range(len(A)), key=A.__getitem__)

Python3实现：

class Solution():
    def maxWidthRamp(self, A):
        res = 0
        m = float('inf')
        for i in sorted(range(len(A)), key=A.__getitem__):
            res = max(res, i - m)
            m = min(m, i)
        return res


if __name__ == '__main__':
    solu = Solution()
    A = [6, 0, 8, 2, 1, 5]
    print(solu.maxWidthRamp(A))

声明： 总结学习，有问题或不当之处，可以批评指正哦，谢谢。

题目链接、代码参考链接

点赞