LeetCode977. 有序数组的平方

一、题目

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

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

示例 2：

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

提示：

• 1 <= nums.length <= 104
• -104 <= nums[i] <= 104
• nums 已按 非递减顺序 排序

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、思路及代码

方法一：暴力求解法

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

方法二：双指针法

（1）因为数组已经排序好了，平方最大的数一定是在两边，我们不妨用双指针来遍历。

（2）比较左右指针指向值的平方，大的放到新建数组的最后面，然后双指针不断向中间靠拢，从而填充新建数组。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> result(nums.size(),0);
        int left = 0;
        int right = nums.size() - 1;
        int k = nums.size() - 1;
        while(left <= right){
            if(pow(nums[left],2)<pow(nums[right],2)){
                result[k--] = pow(nums[right--],2);
            }else{
                result[k--] = pow(nums[left++],2);
            }
        }
        return result;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17