解法一

先倒序排序并存储排序前的序号。用一个指针指向排序后数组的第一个数,然后开始计算答案数组。
如果当前指向的这个元素的原始下标小于等于当前答案数组元素的下标,说明这个元素原来在左侧,因此指针后移,考虑下一个次大值,否则它就是右侧的最大值。
时间复杂度:1299. 将每个元素替换为右侧最大元素 - 图1, N为数组长度。

  1. import java.util.Arrays;
  2. import java.util.Comparator;
  3. class Solution {
  4. Comparator<Element> elementComparator = new Comparator<Element>() {
  5. @Override
  6. public int compare(Element o1, Element o2) {
  7. // 按照value的值降序排列
  8. return (o2.value - o1.value);
  9. }
  10. };
  11. public int[] replaceElements(int[] arr) {
  12. Element[] elements = new Element[arr.length];
  13. for (int i = 0; i < arr.length; ++i) {
  14. elements[i] = new Element(arr[i], i);
  15. }
  16. Arrays.sort(elements, elementComparator);
  17. int[] ans = new int[arr.length];
  18. ans[arr.length - 1] = -1;
  19. int ptr = 0;
  20. for (int i = 0; i < arr.length - 1; ++i) {
  21. while ((ptr < arr.length) && (elements[ptr].index <= i)) {
  22. ++ptr;
  23. }
  24. ans[i] = elements[ptr].value;
  25. }
  26. return ans;
  27. }
  28. }
  29. class Element {
  30. // 元素值
  31. public int value;
  32. // 元素序号
  33. public int index;
  34. public Element(int value, int index) {
  35. this.value = value;
  36. this.index = index;
  37. }
  38. }

解法二

从后向前更新最大值,最后一个数特殊处理。
时间复杂度:1299. 将每个元素替换为右侧最大元素 - 图2, N为数组长度。

  1. class Solution {
  2. public int[] replaceElements(int[] arr) {
  3. int[] ans = new int[arr.length];
  4. ans[arr.length - 1] = arr[arr.length - 1];
  5. int max = -1;
  6. for (int i = arr.length - 2; i >= 0; --i) {
  7. max = Math.max(max, arr[i + 1]);
  8. ans[i] = max;
  9. }
  10. ans[arr.length - 1] = -1;
  11. return ans;
  12. }
  13. }