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

在这里插入图片描述
给你一个数组 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] <= 10^5

Code:

class Solution {
public:
    vector<int> replaceElements(vector<int>& arr) {
        vector<int>res;
        for(int i=0;i<arr.size();i++)
        {
            if(i==arr.size()-1)
                res.push_back(-1);
            else
                res.push_back(*max_element(arr.begin()+i+1,arr.end()));
        }
        return res;

    }
};
posted @ 2022-05-15 10:47  萧海~  阅读(6)  评论(0编辑  收藏  举报