原题链接:点击这里
题目描述:
给你一个按 非递减顺序 排序的整数数组 nums
,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
思路:
题目让我们使用双指针来解决问题,同时我们注意到数组nums的平方最大值一定存在于两端中的一个,所以可以考虑用a指向首地址,用b指向末地址,然后创建一个新数组res,并用index指向末地址,每次比较原数字两端的平方值,将较大者填入新数组末端。如若nums[a]^2>nums[b]^2,则res[index]=nums[a]^2;index--;a++;
复杂度分析:
时间复杂度:O(n);
空间复杂度:O(2n);
AC代码:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> res(nums.size());
int a=0, b=nums.size()-1, index=b;
while(a<=b)
{
if(pow(nums[a],2)>pow(nums[b],2))
{
res[index]=pow(nums[a],2);
index--;
a++;
}
else
{
res[index]=pow(nums[b],2);
index--;
b--;
}
}
return res;
}
};