[LeetCode] 每日一题 1299. 将每个元素替换为右侧最大元素
题目链接
题目描述
给你一个数组 arr
,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1
替换。
完成所有替换操作后,请你返回这个数组。
示例输入
示例 1
输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]
解释:
- 下标 0 的元素 --> 右侧最大元素是下标 1 的元素 (18)
- 下标 1 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 2 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 3 的元素 --> 右侧最大元素是下标 4 的元素 (6)
- 下标 4 的元素 --> 右侧最大元素是下标 5 的元素 (1)
- 下标 5 的元素 --> 右侧没有其他元素,替换为 -1
示例 2
输入:arr = [400]
输出:[-1]
解释:下标 0 的元素右侧没有其他元素。
提示
1 <= arr.length <= 10^4
1 <= arr[i] <= 10^5
解题思路
今天这道题是一个相对简单的题目,要求将数组中的每个元素替换为它右边最大的元素。如果是最后一个元素,则替换为 -1
。这类问题可以通过从右到左遍历数组来高效解决
我的思路是:从数组的右边开始遍历,逐个元素处理。首先,我们需要用一个变量来存储当前遍历到的右侧最大的元素,并从数组的最后一个元素开始,将其值设为 -1
。然后,从倒数第二个元素开始向左遍历,对于每个元素,更新其值为右边最大元素,并更新右边最大值
这种做法的关键在于从右向左遍历数组,这样可以确保每次都能用右边的最大值来替换当前元素
代码实现
class Solution {
public int[] replaceElements(int[] arr) {
int n = arr.length;
int max = arr[n - 1];
arr[n - 1] = -1;
for (int i = n - 2; i >= 0; i--) {
int temp = arr[i];
arr[i] = max;
max = Math.max(temp, max);
}
return arr;
}
}
复杂度分析
时间复杂度:我们只遍历了数组一次,时间复杂度是 O(n),其中 n 是数组的长度
空间复杂度:我们只用了常数空间来存储当前的最大值,因此空间复杂度是 O(1)
总结
这道题目考察了如何通过遍历数组并维护一个最大值来解决问题。通过从右到左的遍历,我们能够在一个遍历过程中完成所有元素的替换。这种方法时间复杂度较低,空间复杂度也是最优的。总的来说,逻辑简单且高效,适合用来巩固对数组操作的理解
希望这篇分享能为你带来启发!如果你有任何问题或建议,欢迎在评论区留言,与我共同交流探讨。
License:
CC BY 4.0