leetcode1299. 将每个元素替换为右侧最大元素

给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1 替换。

完成所有替换操作后,请你返回这个数组。

 

示例:

输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]


 

提示:


    1 <= arr.length <= 10^4
    1 <= arr[i] <= 10^5

逆序遍历,记录之前答案,和当前元素比对即可。

class Solution:
    def replaceElements(self, arr: List[int]) -> List[int]:
        n = len(arr)
        ans = [0] * (n - 1) + [-1]
        for i in range(n - 2, -1, -1):
            ans[i] = max(ans[i + 1], arr[i + 1])
        return ans

 

©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页