今天刷有序数组的平方,大家有兴趣可以点上看看题目要求,试着做一下。
方法一,直接排序,直接平方赋给一个新数组,再排序(函数或者其他排序算法冒泡之类的)。
方法二,双指针,两个指针同时从左往右,或者一个在左一个在右往中间遍历(最符合题意)。
我们直接看题解吧:
审题目+事例+提示:
·nums是已经排好序的数组(负数正数都可能有)。
·非递减顺序,即类似升序,但存在重复元素
思路(双指针):
从左右两边进行比较,
此时数组元素有三种情况,
第一种全为正数,第二种全为负数,
这两种都可以比较也好理解。
第三种正负皆有,此时平方后各元素大小关系类似于y=x2的函数图像两头大中间小,因此也可以比较。
思路步骤:
定义le与ri两个指针,以及新数组指针pos,然后取元素平方比较,较大的逆序(即放最右边)放入新数组,重复上述步骤直至两指针指向同一位置。
代码:
class Solution {
public int[] sortedSquares(int[] nums) {
int n = nums.length;
int[] ans = new int[n];
for (int i = 0, j = n - 1, pos = n - 1; i <= j;) {
if (nums[i] * nums[i] > nums[j] * nums[j]) {
ans[pos] = nums[i] * nums[i];
++i;
} else {
ans[pos] = nums[j] * nums[j];
--j;
}
--pos;
}
return ans;
}
}
一开始有想到双指针,但是没理解题目说数组说已排好序,然后就想怎么不用创建新数组原地平方+排序