双指针
题目
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
实例
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
思路
我们可以使用两个指针分别指向位置 0和 n-1,
每次比较两个指针对应的数,
选择较大的那个逆序放入答案并移动指针。
代码
class Solution {
public int[] sortedSquares(int[] nums) {
int n=nums.length;
int[] ans= new int[n];
for(int i=0,j=n-1,pos=n-1;i<=j;){
if(nums[i]*nums[i]<nums[j]*nums[j]){
ans[pos]=nums[j]*nums[j];
--j;
}else{
ans[pos]=nums[i]*nums[i];
++i;
}
--pos;
}
return ans;
}
}