LeetCode题解之将每个元素替换为右侧最大元素

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

题目描述

给你一个数组 arr ，请你将每个元素用它右边最大的元素替换，如果是最后一个元素，用 -1 替换。

完成所有替换操作后，请你返回这个数组。

示例 :

输入：arr = [17,18,5,4,6,1]
输出：[18,6,6,6,1,-1]

提示：

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

解题思路

从数组最后遍历，每次记录数组右侧最大元素并与当前元素进行比较替换。

复杂度分析

• 时间复杂度：O(n)，其中 n 为数组 arr 的长度 。
• 空间复杂度：O(1)。

代码实现

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(res[i+1], arr[i+1]);
        }
        return res;
    }
}