题目
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入: nums = [-4,-1,0,3,10]
输出: [0,1,9,16,100]
解释: 平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
题解
方式一:暴力
public int[] sortedSquares(int[] nums) {
int[] result = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
result[i] = nums[i] * nums[i];
}
Arrays.sort(result);
return result;
}
方式二:双指针
原数组有序,所以平方后相对有序
public int[] sortedSquares(int[] nums) {
int n = nums.length;
int mix = 0; // 平方最小的下标
int[] arr = new int[n]; // 原数组平方
for (int i = 0; i < nums.length; i++) {
int tem = nums[i] * nums[i];
mix = tem < arr[mix] ? i : mix;
arr[i] = tem;
}
int[] result = new int[n];
result[0] = arr[mix];
int l = mix - 1, r = mix + 1, index = 1;
while (l >= 0 && r < n) {
if (arr[l] > arr[r]) {
result[index++] = arr[r++];
} else {
result[index++] = arr[l--];
}
}
while (index < n) {
if (l < 0) {
result[index++] = arr[r++];
} else {
result[index++] = arr[l--];
}
}
return result;
}
总结
算法:双指针