文章

[LeetCode] 每日一题 1299. 将每个元素替换为右侧最大元素

题目链接

https://leetcode.cn/problems/replace-elements-with-greatest-element-on-right-side

题目描述

给你一个数组 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