给你一个按 非递减顺序 排序的整数数组 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]
实现代码:
暴力法
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums)
{
for(int i=0;i<nums.size();i++)
{
nums[i]*=nums[i];
}
sort(nums.begin(),nums.end());
return nums;
}
};
直接平方后排序即可。
- 时间复杂度:O(n + nlogn)
双指针法
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums)
{
vector<int> v(nums.size());
int i=0;
int j=nums.size()-1;
int k=nums.size()-1;//新数组的下标
while(i<=j)
{
if(nums[i]*nums[i]<nums[j]*nums[j])
{
v[k--]=nums[j]*nums[j];
j--;
}
else
{
v[k--]=nums[i]*nums[i];
i++;
}
}
return v;
}
};
根据题意,平方后的数组的元素大小一定是由两侧向中间递减,所以创建一个相同大小的新数组,双指针指向原数组的头和尾,比较大小后,移动较大值的指针向内侧移动,同时将值赋给指向新数组最后一个未赋值的地方。要注意while里的判断条件要带上=号,否则当两个指针指向同一元素的时候,此元素不会进入新数组。
- 时间复杂度:O(n)