难度:简单
给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序
暴力解法
思路:先平方,后排序
var sortedSquares = function(nums) {
for(let i=0;i<nums.length;i++){
nums[i] *=nums[i]
}
nums.sort((a,b)=>
a-b
)
return nums
};
时间复杂度是 O(n + nlogn)
双指针法
思路:数组其实是有序的, 只不过负数平方之后可能成为最大数了。
那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
此时可以考虑双指针法了,i指向起始位置,j指向终止位置。
定义一个新数组result,和A数组一样的大小,让k指向result数组终止位置。
如果nums[i] * nums[i] < nums[j] * nums[j] 那么array[k–] = nums[j] * nums[j]; 。
如果nums[i] * nums[i] >= nums[j] * nums[j] 那么array[k–] = nums[i] * nums[i]; 。
var sortedSquares = function(nums) {
var index = nums.length-1
var array = new Array(index).fill(0)
for(let i=0,j=nums.length-1;i<=j;){ //i为左指针,j为右指针
if(nums[i]*nums[i]<nums[j]*nums[j]){ //右指针指向的值nums[j]*nums[j]是最大的数
array[index]=nums[j]*nums[j]
index--
j-- //右指针向前移一位
}else{ //左指针指向的值nums[i]*nums[i]是最大的数
array[index]=nums[i]*nums[i]
index--
i++ //左指针向后移一位
}
}
return array
};
时间复杂度为O(n)