给你一个按 非递减顺序 排序的整数数组 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) 的算法解决本问题
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
- 方法一:
查找临界点,因为已经是按照递增的顺序排序,按照临界点,右边按照顺序的方式排列,左边按照逆序的方式排列,这样就可以用归并的思想来完成,找出最后一个负数为临界点,如果临界点的平方大于临界点+1的平方,那么临界点+1最为最小值,将它填入到临时数组的起点,然后临时数组的下标+1,因为临界点最左边的数的平方都比临界点大,所以向右移动临界点+1的下标继续比较;而如果临界点的平方小于临界点+1的平方,就将临界点添加到数组中,然后将临界点向左移动继续比较;
如果临界点不存在,说明数组全是正数,直接平方填入即可;
如果临界点=数组长度-1,说明全是负数,按照逆序填入平方即可
public static int[] sortedSquares(int[] nums) {
int n = nums.length;
int[] result = new int[nums.length];
int side=-1;
for(int i=0;i<n;++i){
if(nums[i]<0){
side=i;
}else{
break;
}
}
int index=0,i=side,j=side+1;
while(i>=0||j<n){
if(i<0){
result[index]=nums[j]*nums[j];
++j;
}else if(j==n){
result[index]=nums[i]*nums[i];
--i;
}else if(nums[i]*nums[i]<=nums[j]*nums[j]){
result[index]=nums[i]*nums[i];
--i;
}else{
result[index]=nums[j]*nums[j];
++j;
}
++index;
}
return result;
}
- 方法二:
用双指针的方法来收缩下标,如果有正有负,那么数组最大值一定存在两端,比较最大值,如果left>rigth,就将nums[left]平方填入到临时数组的最高位,然后left++向右移动,如果小于,就将nums[right]的平方值填入到临时数组的最高位,然后right–向左移动,同时两者都要将临时数组的指针向左移动,一直查找直到left>right为止.
如果全为正或者负,那么就其中一个指针移动直到两个指针重合,同样从临时数组的最高位开始填.
public static int[] sortedSquares2(int[] nums) {
int left = 0;
int right = nums.length-1;
int[] result=new int[right+1];
int write = right;
while(left<=right){
if(nums[left]*nums[left]>nums[right]*nums[right]){
result[write]=nums[left]*nums[left];
++left;
}else{
result[write]=nums[right]*nums[right];
--right;
}
--write;
}
return result;
}