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

120 阅读2分钟

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

问题描述

给你一个数组 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、最后一个元素使用-1替代

最后一个元素使用-1替代,并初始化当前最大值。

let max = arr[arr.length - 1];
arr[arr.length - 1] = -1;
  • 2、从右往左扫描
for(let i = arr.length - 2;i >= 0; i--){
    ……
}
  • 3、将当前扫描位置的元素使用记录的最大值代替
for(let i = arr.length - 2;i >= 0; i--){
    ……
    arr[i] = max;
    ……
}
  • 4、更新最大值

在元素使用最多值代替前,我们需要使用一个中间变量来先将原本的元素值记录下来。

for(let i = arr.length - 2;i >= 0; i--){
    const temp = arr[i];
    ……
    max = Math.max(temp,max);
}

AC 代码

完整代码如下:

/**
 * @param {number[]} arr
 * @return {number[]}
 */
var replaceElements = function(arr) {
    let max = arr[arr.length - 1];
    arr[arr.length - 1] = -1;
    for(let i = arr.length - 2;i >= 0; i--){
        const temp = arr[i];
        arr[i] = max;
        max = Math.max(temp,max);
    }
    return arr;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。