Leetcode977

有序数组的平方

Leetcode977

​ 给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。

示例 1:

1
2
3
4
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]

示例 2:

1
2
输入:nums = [-7,-3,2,3,11]
输出:[4,9,9,49,121]

暴力法

​ 最简单的当然就是暴力法,直接平方,再排序。

1
2
3
4
5
6
7
8
9
10
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
for(int i = 0; i < nums.size(); i++){
nums[i] = nums[i] * nums[i];
}
sort(nums.begin(),nums.end());
return nums;
}
};

双指针法

​ 已知数组是有序的,非递减排序,只不过负数平方之后,可能比一些数要大得多,例如$ (-4)^2 > 1^2$ ,从而影响排列顺序。
重点:数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
​ 因此,这里考虑使用一个双指针,一个指向数组第一项——i,一个指向数组最后一项——j。
​ 定义一个新数组result,和nums数组一样的大小,让k指向result数组终止位置
​ 当前面比后面大时,直接将其赋给新数组的最后一项,i++;
​ 当后面比前面大时,直接将其赋给新数组的最后一项,j–;

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int left = 0; //左边指针
int right = nums.size()-1; //右边指针
vector<int> result(nums.size());
int k = right; //指向result数组最后一个元素
while(left <= right){
if(nums[left] * nums[left] < nums[right]*nums[right]){
result[k--] = nums[right]*nums[right];
right--;
}
else{
result[k--] = nums[left]*nums[left];
left++;
}
}
return result;
}
};
  • Copyright: Copyright is owned by the author. For commercial reprints, please contact the author for authorization. For non-commercial reprints, please indicate the source.
  • Copyrights © 2019-2022 1nvisble
  • Visitors: | Views:

请我喝杯咖啡吧~

支付宝
微信