题目
题目传送门(点击即可)
给你一个按 非递减顺序 排序的整数数组 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:暴力法
暴力法很简单,大部分人几乎都能想到。就是计算完每个每个数的平方,在对其进行排序即可。这样时间复杂度为 O ( n 2 ) O(n^2) O(n2)。由于这种方法过于简单,就不在赘述。下面主要讲指双针法。
法2:双指针法
由于原始数组即有序,因此平方后的数组唯一使顺序变动的地方就是负数导致的,即负数越小其平方反而越大。因此最小值一定在正负交界处,最大值一定在数组的两端。
因此我们可以考虑从交界处出发,同时向两端靠,依次比较元素。(这里是从中间向两端出发,也可以从两端向中间搜索)。为此设定两个指针,一个指向最大负值,另一个指向0或最小正数(显然两指针应该是挨着)。然后分别向两头遍历,依次比较元素,小的就插入且指针后移。后移后继续比较。
以上就是大致思路,这样做实现复杂度压缩为了 O ( n 2 ) O(n^2) O(n2)。java版本代码如下:
int positivePtr = nums.length;//默认为数组长度,这是在数组全为负数的情况下
int result[] = new int[nums.length]; //存放结果
for (int i = 0; i < nums.length; i++) //找到第一个非负数
if(nums[i]>=0){
positivePtr = i;
break;
}
int negativePtr = positivePtr-1;
int index = 0;
while(negativePtr>=0&&positivePtr< nums.length){
if(-nums[negativePtr]<=nums[positivePtr]){//
result[index++] = nums[negativePtr]*nums[negativePtr];
negativePtr--;
}
else{
result[index++] = nums[positivePtr]*nums[positivePtr];
positivePtr++;
}
}
while (positivePtr< nums.length) {//还有正数没处理完
result[index++] = nums[positivePtr]*nums[positivePtr];
positivePtr++;
}
while (negativePtr>=0){
result[index++] = nums[negativePtr]*nums[negativePtr];
negativePtr--;
}
return result;