【刷题笔记】1299. 将每个元素替换为右侧最大元素

26 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第32天,点击查看活动详情

题目描述

1299. 将每个元素替换为右侧最大元素 - 力扣(LeetCode)

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

思路分析

从题目可以看出

  • 从右到左遍历的话,刚好是从遍历过的数中去寻找最大值。
  • 从左到右遍历的话,那么就需要提前比较还未遍历的值。

不难看出从右到左遍历只需要O(n)时间复杂度,

从左到右遍历需要O(n^2)的时间复杂度

在这个题感悟到,其实有时候生活中遇到困难了,不妨换个思维去想一下

也许当前困难不过是对自己的一种考验

多花点思维思考去解决比用简单的惰性思维去“懒”解决要为后续实现的节省更多时间

AC 代码

class Solution {
    public int[] replaceElements(int[] arr) {
        int max = -1;
        for (int i = arr.length - 1; i >= 0; i--) {
            int tmp = arr[i];
            arr[i] = max;
            if (tmp > max) {
                max = tmp;
            }
        }
        return arr;
    }
}

参考

Java 百分百 - 将每个元素替换为右侧最大元素 - 力扣(LeetCode)

从右往左一次遍历 - 将每个元素替换为右侧最大元素 - 力扣(LeetCode)