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

目录结构

1.题目

2.题解


1.题目

给你一个数组 arr ,请你将每个元素用它右边最大的元素替换,如果是最后一个元素,用 -1 替换。

完成所有替换操作后,请你返回这个数组。

示例:

输入:arr = [17,18,5,4,6,1]
输出:[18,6,6,6,1,-1]

提示:

  • 1 <= arr.length <= 10^4
  • 1 <= arr[i] <= 10^5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/replace-elements-with-greatest-element-on-right-side
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2.题解

逆序地遍历数组,同时维护从数组右端到当前位置所有元素的最大值。

public class Solution1299 {
    public int[] replaceElements(int[] arr) {
        int len = arr.length, tmpMax = arr[len - 1];
        int[] result = new int[len];
        result[len - 1] = -1;
        for (int i = len - 2; i >= 0; i--) {
            result[i] = tmpMax;
            if (arr[i] > tmpMax) {
                tmpMax = arr[i];
            }
        }
        return result;
    }
}
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

你可能感兴趣的:(LeetCode)