题目
给你一个按 非递减顺序 排序的整数数组 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]
思路:
- 可以直接平方插入新数组在排序。
- 使用双指针,由于数组是中包含负数和正数,且是有序的,我们可以寻找负数和正数的边界,此时往左平方与往右平方都是递增的,这时我们可以定义双向指针标记数组中间,比较左指针和右指针的大小,将小的插入数组即可。需要注意边界问题,如果有一边已经到达了边界,则插入另一边的平方即可,
- 根据2的思路可以将指针的位置置为0和nums.length-1,向中间遍历,此时不用寻找负数和正数的边界值,且不用考虑左右边界问题。
//这里我采用方法3
package Leecode;
import java.util.Arrays;
public class test007 {
public static void main(String[] args) {
int[] a = {-4, -1, 0, 3, 10};
int[] b = sortedSquares(a);
System.out.println(Arrays.toString(b));
}
public static int[] sortedSquares(int[] nums) {
int left = 0, right = nums.length - 1;
int index = nums.length - 1;
int[] res = new int[nums.length];
while (left <= right) {
if (left == right) {
res[index] = nums[index] * nums[index];
}
if ((nums[left] * nums[left]) > (nums[right] * nums[right])) {
res[index] = nums[left] * nums[left];
left++;
} else {
res[index] = nums[right] * nums[right];
right--;
}
index--;
}
return res;
}
}