数组 | 977. 有序数组的平方

977. 有序数组的平方

参考：
代码随想录：双指针法

零、描述

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

举例：

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

解法：双指针

参考代码：

class Solution {
    
    
public:
    vector<int> sortedSquares(vector<int>& A) {
    
    
        int k = A.size() - 1;
        vector<int> result(A.size(), 0);
        for (int i = 0, j = A.size() - 1; i <= j;) {
    
     // 注意这里要i <= j，因为最后要处理两个元素
            if (A[i] * A[i] < A[j] * A[j])  {
    
    
                result[k--] = A[j] * A[j];
                j--;
            }
            else {
    
    
                result[k--] = A[i] * A[i];
                i++;
            }
        }
        return result;
    }
};

自己的：

class Solution {
    
    
public:
    vector<int> sortedSquares(vector<int>& nums) {
    
    
        //想法：双指针，创建新的数组，左右指针对比
        int N=nums.size();
        vector<int> res(N);
        int l=0,r=N-1;
        int i=N-1;
        while(l<=r){
    
    
            int l2=nums[l]*nums[l];
            int r2=nums[r]*nums[r];
            if(l2<r2){
    
    
                res[i]=r2;
                r--;
            }
            else{
    
    
                res[i]=l2;
                l++;
            }
            i--;
        }
        return res;
    }
};