用 C++ 替换元素,使其右侧成为最大元素
假设我们有一个数组 A。我们必须用这个元素右侧的最大元素替换每个元素。并将最后一个替换为 -1。所以如果 A = [5, 17, 40, 6, 3, 8, 2],它将变成 [40,40,8,8,8,2,-1]
为了解决这个问题,我们将遵循以下步骤 −
- 我们将从右向左读取数组元素。
- 取 e := -1
- 对于 i := n – 1 至 0
- temp := e
- e := e 与 array[i] 之间的最大值
- array[i] := temp
- 返回 array
示例
让我们看看以下实现,以获得更好的理解 −
#include <bits/stdc++.h> using namespace std; void print_vector(vector<int> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: vector<int> replaceElements(vector<int>& arr) { int rep = -1; int n = arr.size(); for(int i = n - 1; i >= 0; i--){ int temp = rep; rep = max(rep, arr[i]); arr[i] = temp; } return arr; } }; main(){ Solution ob; vector<int> c = {5,17,40,6,3,8,2}; print_vector(ob.replaceElements(c)) ; }
输入
[5,17,40,6,3,8,2]
输出
[40,40,8,8,8,2,-1]
广告