给你一个数组 `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`
首先这道题并不难,但是我个人的方法比较局限,所以扩展一下个人的思维;
个人所写
class Solution {
public:
vector<int> replaceElements(vector<int>& arr) {
for(int i=0;i<arr.size();i++)
{
arr[i]=max_back(i,arr);
}return arr;
}
int max_back(int start,vector<int>& arr)
{
if(start==arr.size()-1)return -1;
int max=arr[start+1];
for (size_t i = start+2; i < arr.size(); i++)
{
if(max<arr[i])max=arr[i];
}
return max;
}
};
上述代码直属暴力,但是在写这道题的时候,感觉有可能超时,事实是,没超时,但是===>
耗时太大,必须优化
以下是对原代码优化:
class Solution {
public:
vector<int> replaceElements(vector<int>& arr) {
int max;
for(int i=0;i<arr.size();i++)
{//下面的代码就是说,先从i+1--n找到了一个最大值max,并将这个最大值赋值给arr[i],而对于下一次循环,如果arr[i+1]<max,也即是说最大的那个值仍然没有被arr[i+1]给代替掉,还是在i+2--n之间,那么就直接将最大值再一次赋值给arr[i+1],一遍一遍的循环
if(i&&arr[i]<arr[i-1])
arr[i]=arr[i-1];
else
arr[i]=max_back(i,arr);
}return arr;
}
int max_back(int start,vector<int>& arr)
{
if(start==arr.size()-1)return -1;
int max=arr[start+1];
for (size_t i = start+2; i < arr.size(); i++)
{
if(max<arr[i])max=arr[i];
}
return max;
}
这一次对于时间复杂度大大的优化
别人所写
从另一个角度想,每次都从i+1–n处寻找最大值,也就是从许多个里面寻找一个最大值,那么能不能先从一个里面寻找一个最大值呢?
本题等价于对于数组 arr 中的每个元素 arr[i],将其替换成 arr[i + 1], arr[i + 2], …, arr[n - 1] 中的最大值。因此我们可以逆序地遍历整个数组,同时维护从数组右端到当前位置所有元素的最大值。
设 ans[i] = max(arr[i + 1], arr[i + 2], …, arr[n - 1]),那么在进行逆序遍历时,我们可以直接通过
ans[i] = max(ans[i + 1], arr[i + 1])
来递推地得到答案。
这种方法很有技巧性,相当于是在倒着推;
class Solution {
public:
vector<int> replaceElements(vector<int>& arr) {
int n = arr.size();
vector<int> ans(n);
ans[n - 1] = -1;
for (int i = n - 2; i >= 0; --i) {
ans[i] = max(ans[i + 1], arr[i + 1]);
}
return ans;
}
};
(int i = n - 2; i >= 0; --i) {
ans[i] = max(ans[i + 1], arr[i + 1]);
}
return ans;
}
};
借鉴于[此处](https://leetcode-cn.com/problems/replace-elements-with-greatest-element-on-right-side/solution/jiang-mei-ge-yuan-su-ti-huan-wei-you-ce-zui-da-y-5/)