概述
分析
- 给出的数组按非递减排序,要求返回 每个数字的平方 组成的新数组,并且新数组也要按非递减排序
- 根据题目下面的示例,我们可以知道,数组中存在正值和负值
- 虽然无法确定原数组元素平方后的最小值,但是我们容易得到最大值一定在原数组的两端,所以我们可以利用双指针,一个指向首部,一个指向尾部,比较这两个元素平方后的大小,来确定平方后的数组的最大值,之后一直按照这个思路进行就行,直到两个指针交错
思路
双指针实现思路?
见代码部分
代码
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> res(nums.size()); // 因为要通过[]直接赋值,所以必须先分配内存
int res_index = nums.size() - 1;
int left_index = 0 , right_index = nums.size() - 1; // 双指针
while (left_index <= right_index) {
if (pow(nums[right_index], 2) > pow(nums[left_index], 2)) // 判断哪一端是最大元素
res[res_index--] = pow(nums[right_index--], 2);
else
res[res_index--] = pow(nums[left_index++], 2);
}
return res;
}
};