[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)

总结

这道题目考察了如何通过遍历数组并维护一个最大值来解决问题。通过从右到左的遍历，我们能够在一个遍历过程中完成所有元素的替换。这种方法时间复杂度较低，空间复杂度也是最优的。总的来说，逻辑简单且高效，适合用来巩固对数组操作的理解

希望这篇分享能为你带来启发！如果你有任何问题或建议，欢迎在评论区留言，与我共同交流探讨。