亲宝软件园·资讯

展开

go语言题解LeetCode1299将每个元素替换为右侧最大元素

刘09k11 人气:0

题目描述

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 百分百 - 将每个元素替换为右侧最大元素

逆序遍历罢了

代码

class Solution {
    public int[] replaceElements(int[] arr) {
       int max = -1;
       int[] ans = new int[arr.length];
       for(int i = arr.length-1;i >= 0;i--){
           int temp = arr[i];
           ans[i] = max;
           max = Math.max(max,temp);
       }
       return ans;
    }
}

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

从右往左遍历,先记录右边最大值 rightMax 为最后一个值,向左每次更新 rightMax,使用变量 t 先记住当前 arr[i] 就可以了。

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

加载全部内容

相关教程
猜你喜欢
用户评论