【LeetCode】977. 有序数组的平方（简单）——代码随想录算法训练营Day02

题目链接：977.有序数组的平方

题目描述

给你一个按 非递减顺序 排序的整数数组 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) 的算法解决本问题

文章讲解：代码随想录

视频讲解： 双指针法经典题目 | LeetCode：977.有序数组的平方_哔哩哔哩_bilibili

题解1：暴力法

思路：首相想到的最简单方法就是先把数组中每个元素平方，然后再排序。

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var sortedSquares = function(nums) {
    // 暴力解法，先平方，再排序
    return nums.map((n) => n * n).sort((a, b) => a - b);
};

分析：此解法并没有用到题目中 nums 已按非递减顺序排序的条件，时间复杂度为 O(nlogn)，空间复杂度为 O(n)。还可以针对题目条件来继续优化。

题解2：相向双指针法

思路：因为数组已按非递减顺序排序，所以数组元素两边大中间小。因此可以定义左右两个指针和一个新的数组存储结果，将左右指针指向的位置元素比较，依次将较大的元素从右向左填入结果数组中。

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var sortedSquares = function(nums) {
  const res = new Array(nums.length); // 用于存放结果
  let index = nums.length - 1; // 从右往左填充最大值
  let left = 0, right = nums.length - 1; // 左右指针分别指向数组的两端
  while (left <= right) {
    const lSquare = nums[left] * nums[left], rSquare = nums[right] * nums[right];
    // 比较左右指针指向的值，将较大的值填入结果数组
    if (lSquare >= rSquare) {
      res[index--] = lSquare;
      left++;
    } else {
      res[index--] = rSquare;
      right--;
    }
  }
  return res;
};

分析：此方法只需遍历1次数组，时间复杂度为 O(n)，空间复杂度为 O(1)。

收获

1. 读题时要仔细分析题目中给出条件，根据重要条件来组织解题思路。

2. 再次使用了相向双指针法，对其更加熟悉。