Leetcode 1299.将每个元素替换为右侧最大元素
1 题目描述(Leetcode题目链接)
给你一个数组 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
2 题解
逆序遍历,记录最大值。
class Solution:
def replaceElements(self, arr: List[int]) -> List[int]:
res = [-1]
val = arr[-1]
for i in range(len(arr) - 2, -1, -1):
res.append(val)
val = max(val, arr[i])
return res[::-1]