LeetCode.1299 将每个元素替换为右侧最大元素

135 阅读1分钟

「这是我参与2022首次更文挑战的第30天,活动详情查看:2022首次更文挑战」。

题目描述:

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

给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -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

示例二

输入: arr = [400]
输出: [-1]
解释: 下标 0 的元素右侧没有其他元素。

提示:

  • 1 <= arr.length <= 10^4
  • 1 <= arr[i] <= 10^5

思路分析

暴力双循环

这个也没啥好说的,就是遍历,每个数与他右边的都比较一遍。

AC代码

class Solution {
    fun replaceElements(arr: IntArray): IntArray {
        for (i in 0 until arr.size) {
            if (i == (arr.size - 1)) {
                arr[i] = -1
                break
            }
            var biggestRNumber = arr[i + 1]
            for (j in i + 1 until arr.size - 1) {
                if (biggestRNumber < arr[j + 1]) {
                    biggestRNumber = arr[j + 1]
                }
            }
            arr[i] = biggestRNumber
        }
        return arr
    }
}

逆序遍历

仔细看下题目要求,我们最终得到的是一个逆序数组,因此我们可以逆序遍历。

AC代码

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

参考

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

将每个元素替换为右侧最大元素【Swift】 - 将每个元素替换为右侧最大元素 - 力扣(LeetCode) (leetcode-cn.com)

Python | Kotlin | Java - 将每个元素替换为右侧最大元素 - 力扣(LeetCode) (leetcode-cn.com)