给你一个按 非递减顺序 排序的整数数组 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]
思路:
双指针
我自己变成了3指针,一个是头指针,两个是尾指针,其中一个尾指针用来不断更新遍历,作为循环终止条件。
精髓在于,数组已按大小排列好,平方后的最大数一定在头或者尾,因此每次比较头指针和尾指针的平方数,哪一个大,新的平方数数组的指针(额外的尾指针)索引为这个大的平方数,这个大数的指针便进行更新(小数的指针不更新),然后额外的尾指针也进行更新。
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> res(nums.size(),0);
int i=0,j=nums.size()-1,k=nums.size()-1;
while(k>=0)
{
if(nums[i]*nums[i]<=nums[j]*nums[j])
{
res[k]=nums[j]*nums[j];
j--,k--;
}
else
{
res[k]=nums[i]*nums[i];
i++,k--;
}
}
return res;
}
};