LeetCode 1299.将每个元素替换为右侧最大元素：倒序遍历，维护最大值，原地修改

【LetMeFly】1299.将每个元素替换为右侧最大元素：倒序遍历，维护最大值，原地修改

力扣题目链接：https://leetcode.cn/problems/replace-elements-with-greatest-element-on-right-side/

给你一个数组 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 <= 104
• 1 <= arr[i] <= 105

解题方法：模拟

使用一个变量M记录右边元素最大值，初始值为-1。

从后向前遍历数组，使用nextM记录遍历完这个元素之后的最大值（nextM = max(arr[i], M)），更新这个元素为M，并更新M为nextM。

最终返回arr即为答案。

• 时间复杂度$O(len(arr))$
• 空间复杂度$O(1)$

AC代码

C++

/*
 * @Author: LetMeFly
 * @Date: 2025-02-16 10:21:04
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-02-16 10:22:18
 */
class Solution {
public:
    vector<int> replaceElements(vector<int>& arr) {
        int M = -1;
        for (int i = arr.size() - 1; i >= 0; i--) {
            int nM = max(M, arr[i]);
            arr[i] = M;
            M = nM;
        }
        return arr;
    }
};

Python

'''
Author: LetMeFly
Date: 2025-02-16 10:21:11
LastEditors: LetMeFly.xyz
LastEditTime: 2025-02-16 10:23:15
'''
from typing import List

class Solution:
    def replaceElements(self, arr: List[int]) -> List[int]:
        M = -1
        for i in range(len(arr) - 1, -1, -1):
            arr[i], M = M, max(arr[i], M)
        return arr

Java

/*
 * @Author: LetMeFly
 * @Date: 2025-02-16 10:21:15
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-02-16 10:24:13
 */
class Solution {
    public int[] replaceElements(int[] arr) {
        int M = -1;
        for (int i = arr.length - 1; i >= 0; i--) {
            int nM = Math.max(M, arr[i]);
            arr[i] = M;
            M = nM;
        }
        return arr;
    }
}

Go

/*
 * @Author: LetMeFly
 * @Date: 2025-02-16 10:21:19
 * @LastEditors: LetMeFly.xyz
 * @LastEditTime: 2025-02-16 10:26:54
 */
package main

func max_REWGEORS(a, b int) int {
    if a > b {
        return a
    }
    return b
}

func replaceElements(arr []int) []int {
    M := -1
    for i := len(arr) - 1; i >= 0; i-- {
        nM := max_REWGEORS(arr[i], M)
        arr[i] = M
        M = nM
    }
    return arr
}

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源

Tisfy：https://blog.letmefly.xyz/2025/02/16/LeetCode 1299.将每个元素替换为右侧最大元素/