题目
977.有序数组的平方
给你一个按 非递减顺序 排序的整数数组 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)
的算法解决本问题
提示:部分内容引用代码随想录
一、分析题目
1.暴力
本题暴力解法显然易见,将原数组的每个元素平方后再进行一次排序就OK了。
2.双指针
数组其实是有序的, 只不过负数平方之后可能成为最大数了。
那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
此时可以考虑双指针法了,i指向起始位置,j指向终止位置。
定义一个新数组result,和A数组一样的大小,让k指向result数组终止位置。
如果A[i] * A[i] < A[j] * A[j]
那么result[k--] = A[j] * A[j];
。
如果A[i] * A[i] >= A[j] * A[j]
那么result[k--] = A[i] * A[i];
。
这将使代码的时间复杂度优化至O(n)
二、使用步骤
1.暴力解法
代码如下(示例):
import java.util.Arrays;
class Solution {
public int[] sortedSquares(int[] nums) {
for(int i=0;i<nums.length;i++)nums[i]=nums[i]*nums[i];
Arrays.sort(nums);
return nums;
}
}
2.双指针解法(即进阶解法)
代码如下(示例):
class Solution {
public int[] sortedSquares(int[] nums) {
int left=0;
int right=nums.length-1;
int arr[]=new int[nums.length];
int index=nums.length-1;
while(left<=right){
int tempr=nums[right]*nums[right];
int templ=nums[left]*nums[left];
if(templ>tempr){
arr[index]=templ;
left++;
}else{
arr[index]=tempr;
right--;
}
index--;
}
return arr;
}
}