LeetCode刷题之1299. 将每个元素替换为右侧最大元素
我不知道将去向何方,但我已在路上! |
---|
时光匆匆,虽未曾谋面,却相遇于斯,实在是莫大的缘分,感谢您的到访 ! |
- 题目:
给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1 替换。
完成所有替换操作后,请你返回这个数组。 - 示例:
示例 1 :
输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]
- 提示:
- 1 <= arr.length <= 10^4
- 1 <= arr[i] <= 10^5
- 代码1:
class Solution:
def replaceElements(self, arr: List[int]) -> List[int]:
a = []
for i in range(len(arr)-1):
a.append(max(arr[i+1:]))
a.append(-1)
return a
- 算法说明:
建立空列表 a,逐个遍历 arr 中的元素,在大于当前索引的元素中,求出最大值,添加到 a 中,遍历到最后一个元素,在 a 中添加 -1,返回 a 即可。 - 代码2:
class Solution:
def replaceElements(self, arr: List[int]) -> List[int]:
max_val = -1
for i in range(len(arr)-1,-1,-1):
temp = arr[i]
arr[i] = max_val
if temp > max_val:
max_val = temp
return arr
- 算法说明:
将最大值设置为 max_val = -1,倒序遍历 arr 逐步更新最大值;temp = arr[i] 记下当前位置的元素,arr[i] = max_val 将当前元素替换为右边最大的元素 max_val,如果当前元素 temp 大于 右边最大的元素 max_val,将 max_val 的值更新为 temp,如果当前元素小于 max_val,保持 max_val 的值不变,继续循环,将下一个元素替换为 max_val。具体过程如下: