给你一个数组 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]:
max_right = -1
for i in range(len(arr) - 1, -1, -1):
arr[i], max_right = max_right, max(max_right, arr[i])
return arr
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!