LeetCode 977题 -> 有序数组的平方
1. 题目描述
-
给一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
-
示例
输入:nums = [-4,-1,0,3,10] 输出:[0,1,9,16,100] 输入:nums = [-7,-3,2,3,11] 输出:[4,9,9,49,121]
2. 解题思路
- 思路1:先计算平方结果,再直接排序
- 思路2:双指针
3. 代码实现
/**
* 1.直接排序
* 先计算平方后直接排序
* @param nums
* @return
*/
public static 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;
}
/**
* 2.双指针
* @param nums
* @return
*/
public static int[] sortedSquaresByPointers(int[] nums){
int[] res = new int[nums.length];
int left = 0;
int right = nums.length - 1;
int index = nums.length - 1;
while (left <= right){
int l = nums[left] * nums[left];
int r = nums[right] * nums[right];
if(l < r){
res[index--] = r;
right--;
}else {
res[index--] = l;
left++;
}
}
return res;
}