题目描述:
给你一个按 非递减顺序 排序的整数数组 nums
,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
双指针算法思路:
1.数组原来是有序的,平方过后负数平方会变大。
2.即使负数平方会变大,最大的平方值只会出现在数组两端
3.用双指针指向数组的两端,去找平方后的最大值,放到新数组的末尾,之后不断向中间移动
代码:
public int[] sortedSquares(int[] nums){
int n=nums.length;
int [] ans=new int[n];
//从两端开始遍历,找到平方最大元素放到新数组ans的末尾
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;
}