刷题小能手之数组篇(7)
-
给你一个按 非递减顺序 排序的整数数组 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]
-
提示:
-
1 <= nums.length <= 104
-
-104 <= nums[i] <= 104
-
nums 已按 非递减顺序 排序
-
进阶:
-
请你设计时间复杂度为 O(n) 的算法解决本问题
*/
package arrayAlgorithm;
public class SquareSorting {
//最简单的就是先平方后排序,但是没有用到有序数组这一条件
//归并排序思想
public static int[] sortedSquares(int[] nums) {
if(nums.length==1){
nums[0] = nums[0] * nums[0];
return nums;
}
int cut = -1; //分割正负数
int left, right;
int[] r = new int[nums.length];
for(int i=0; i<nums.length; i++){
if (nums[i]<0){
cut = i;
}
}
left = cut;
right = cut+1;
int n = 0; //新数组下标
while (left >= 0||right < nums.length){
if(left<0){
r[n] = nums[right]*nums[right];
right++;
}else if(right==nums.length){
r[n] = nums[left]*nums[left];
left--;
}else if(nums[left]*nums[left]<=nums[right]*nums[right]){
r[n] = nums[left]*nums[left];
left--;
}else if(nums[left]*nums[left]>nums[right]*nums[right]){
r[n] = nums[right]*nums[right];
right++;
}
n++;
}
return r;
}
public static int[] sortedSquares2(int[] nums) {
int n = nums.length;
int negative = -1;
for (int i = 0; i < n; ++i) {
if (nums[i] < 0) {
negative = i;
} else {
break;
}
}
int[] ans = new int[n];
int index = 0, i = negative, j = negative + 1;
while (i >= 0 || j < n) {
if (i < 0) {
ans[index] = nums[j] * nums[j];
++j;
} else if (j == n) {
ans[index] = nums[i] * nums[i];
--i;
} else if (nums[i] * nums[i] < nums[j] * nums[j]) {
ans[index] = nums[i] * nums[i];
--i;
} else {
ans[index] = nums[j] * nums[j];
++j;
}
++index;
}
return ans;
}
public static void main(String[] args) {
// int[] nums1 = {-4, -1, 0, 3, 10};
// int[] nums2 = {1, 3, 10};
// int[] nums3 = {-4, -1, -3};
//
// System.out.println("-4, -1, 0, 3, 10:");
// for(int x: sortedSquares(nums1)){
// System.out.print(x+", ");
// }
// System.out.println("\n");
//
// System.out.println("1, 3, 10:");
// for(int x: sortedSquares(nums2)){
// System.out.print(x+", ");
// }
// System.out.println("\n");
//
// System.out.println("-4, -1, -3");
// for(int x: sortedSquares(nums3)){
// System.out.print(x+", ");
// }
int[] nums4 = {-5,-3,-2,-1};
System.out.println("[-5,-3,-2,-1]");
for(int x: sortedSquares(nums4)){
System.out.print(x+", ");
}
}
}