题目链接: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) 的算法解决本问题
文章讲解:代码随想录
思路:首相想到的最简单方法就是先把数组中每个元素平方,然后再排序。
/**
* @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)。还可以针对题目条件来继续优化。
思路:因为数组已按非递减顺序排序,所以数组元素两边大中间小。因此可以定义左右两个指针和一个新的数组存储结果,将左右指针指向的位置元素比较,依次将较大的元素从右向左填入结果数组中。
/**
* @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. 再次使用了相向双指针法,对其更加熟悉。