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

x33g5p2x  于2021-09-24 转载在 其他  
字(0.4k)|赞(0)|评价(0)|浏览(302)

题目:

思路:

每个元素都要被它右边最大的元素替换,那么主要问题就是怎么方便的找到其右边最大的元素,这里创建一个结果向量来保存结果数据,先将结果向量的最后一个元素置为 -1,再从右向左操作,当要更新结果向量第 i 个位置的值时,要看到其第 i+1 个位置保存的是原向量第 i+1 个位置之后的元素的最大值,因此直接将结果向量第 i+1个位置的数值与原向量第 i+1 个位置的值比较,其中较大值就是结果向量第 i 个位置的值。

代码实现:

vector<int> replaceElements(vector<int>& arr) {
        int n = arr.size();
        vector<int> res(n);
        ans[n - 1] = -1;
        for (int i = n - 2; i >= 0; --i) {
            res[i] = max(res[i + 1], arr[i + 1]);
        }
        return res;
    }

相关文章