977. 有序数组的平方

题目

🔗题目链接：977. 有序数组的平方 - 力扣（LeetCode）

给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。

示例 1：

输入： nums = [-4,-1,0,3,10]
输出： [0,1,9,16,100]
解释： 平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

示例 2：

输入： nums = [-7,-3,2,3,11]
输出： [4,9,9,49,121]

提示：

• 1 <= nums.length <= 104
• -104 <= nums[i] <= 104
• nums 已按 非递减顺序 排序

进阶：

• 请你设计时间复杂度为 O(n) 的算法解决本问题

思路

给的数组是有序的，并且要求返回的数组也是有序的。

1. 全部平方，然后排序。这种方式非常质朴，就不多解释了
2. 双指针。left = 0， right = nums.length - 1，然后两个指针往中间走。当然，谁指向的元素平方值大谁往中间走。

代码

function sortedSquares(nums: number[]): number[] {
  const lens = nums.length;
  // 左指针
  let left = 0;
  // 右指针
  let right = lens - 1;

  // 结果数组，用于存放平方后比较过的结果
  const stack = new Array<number>(lens).fill(0);
  // 结果数组索引，从后往前给数组元素赋值
  let stackIndex = lens - 1;

  while (left <= right) {
    const leftPow = nums[left] * nums[left];
    const rightPow = nums[right] * nums[right];

    // 谁指向的元素的平方值大，谁指向的元素的平方值进结果数组，谁往中间走
    // 数组是从后往前赋值
    // 如果使用 unshift 函数，那就是把大的值往后推
    if (rightPow > leftPow) {
      stack[stackIndex] = rightPow;
      // stack.unshift(rightPow);
      right--;
    } else {
      stack[stackIndex] = leftPow;
      // stack.unshift(leftPow);
      left++;
    }

    stackIndex--;
  }

  return stack;
};