给你一个按 非递减顺序 排序的整数数组 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+logn),空间复杂度为O(1).
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int len=nums.size();
for(int i=0;i<len;i++){
nums[i]=nums[i]*nums[i];
}
sort(nums.begin(),nums.end());
return nums;
}
};
解法二:双指针
已知nums已按非递减排序,所以平方之后我们不确定的是负数的平方是否会大于正数,所以我们设置两个指针,low指向下标为0,high指向size-1,再创建一个数组ans,平方之后进行比较把每次较大的数放入创建的数组的最后一个位置,后面依次往前。两个数进行比较,如果low大,则low++再进行比较,反之,则high--,知道low=high就退出比较,返回数组ans。其时间复杂度为O(n),空间复杂度为O(n)。
class Solution { public: vector<int> sortedSquares(vector<int>& nums) { int len=nums.size(); int low=0; int high=len-1; int k=high; vector<int> ans(len); while(k>=0){ if(nums[low]*nums[low]<nums[high]*nums[high]){ ans[k]=nums[high]*nums[high]; high--; } else{ ans[k]=nums[low]*nums[low]; low++; } k--; } return ans; } };