有序数组的平方
给你一个按 非递减顺序 排序的整数数组 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]
方法一:直接排序
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
res[i] = nums[i] * nums[i];
}
Arrays.sort(res);
return res;
}
方法二:双指针
根据有序特点,先找到最后一个负数的位置,对于平方,其左边的结果从右往左递增,其右边的结果从左往右递增
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
int lastNegative = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] < 0) {
lastNegative = i;
} else {
break;
}
}
int left = lastNegative, right = lastNegative + 1;
int i = 0;
while (left >= 0 || right < nums.length) {
int mul1 = left >= 0 ? nums[left] * nums[left] : Integer.MAX_VALUE;
int mul2 = right < nums.length ? nums[right] * nums[right] : Integer.MAX_VALUE;
if (mul1 < mul2) {
res[i++] = mul1;
left--;
} else {
res[i++] = mul2;
right++;
}
}
return res;
}
双指针2
数组平方从两边向中间递减
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
int left = 0, right = nums.length - 1;
int i = nums.length - 1;
while (left <= right) {
int mul1 = nums[left] * nums[left];
int mul2 = nums[right] * nums[right];
if (mul1 > mul2) {
res[i--] = mul1;
left++;
} else {
res[i--] = mul2;
right--;
}
}
return res;
}