给你一个按 非递减顺序 排序的整数数组 nums
,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
拿到这题我首先想到的就是平方之后直接排序,简单粗暴。
class Solution {
public int[] sortedSquares(int[] nums) {
int[] a = new int[nums.length];
for (int i = 0; i < nums.length; ++i) {
a[i] = nums[i] * nums[i];
}
Arrays.sort(a);
return a;
}
}
但是这个并没有用到非递减顺序 这一个条件,为了用到这个条件,采用双指针的方法,如果全为正数,双指针也一样适用,但是如果是有正有负,而且是非递减顺序,那么双指针一个从前一个从后,一次遍历,哪个大就放入新数组中。大大提高了效率。
class Solution {
public int[] sortedSquares(int[] nums) {
int n = nums.length;
int[] a = new int[n];
for(int i=0,j=n-1,q=n-1;i<=j;){
if(nums[i]*nums[i]>nums[j]*nums[j]){
a[q]=nums[i]*nums[i];
i++;
}else{
a[q]=nums[j]*nums[j];
j--;
}
q--;
}
return a;
}
}