给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1 替换。
完成所有替换操作后,请你返回这个数组。
示例:
输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]
思路:右边最大的元素,考虑从最后的元素开始替换,设置max,max比当前元素小时,将max设置为当前元素
class Solution:
def replaceElements(self, arr: List[int]) -> List[int]:
max = -1
j = len(arr)-1
for i in range(j, -1, -1):
if arr[i] >= max:
arr[i], max = max, arr[i]
else:
arr[i] = max
return arr