leetcode 977. 有序数组的平方

leetcode 977. 有序数组的平方

题目描述：

给你一个按 非递减顺序 排序的整数数组 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]

解题分析：

方法一：暴力解法

首先就是暴力解法，直接先将数组中的元素都进行平方，然后再升序就可以。

时间复杂度：O(nlogn)。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n = nums.size();
        for (int i = 0; i < n; i++){
            nums[i] *= nums[i];
        }
        sort(nums.begin(),nums.end());
        return nums;
    }
};

方法二：双指针法

根据题意，原数组是升序的， 只不过有负数的存在，而题目要求我们生成一个平方数组，从小到大排好序返回；那么数组中元素平方的最大值就在原数组的最左边或者最右边。
那么就可以定义一个新数组 q ，和原数组的大小一样，然后使用双指针法，i 指向起始位置，j 指向终止位置。比较首尾元素平方的大小，将最大值放入平方数组的最后一个位置，再慢慢倒着向数组中添加元素。

时间复杂度： O(n)。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n = nums.size(), k = n - 1;
        vector<int> q(n);
        for (int i = 0, j = n - 1; i <= j;){
            if (nums[i] * nums[i] > nums[j] * nums[j]){
                q[k--] = nums[i] * nums[i];
                i++;
            }
            else{
                q[k--] = nums[j] * nums[j];
                j--;
            }
        }
        return q;
    }
};