力扣 有序数组的平方

力扣 有序数组的平方

题目描述

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

示例 1：

输入：[-4,-1,0,3,10]
输出：[0,1,9,16,100]
示例 2：

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

提示：

1 <= A.length <= 10000
-10000 <= A[i] <= 10000
A 已按非递减顺序排序。

要点分析

方法一: 模拟

时间复杂度:O(nlog(n))

空间复杂度:O(log(n))

AC代码:

#include<algorithm>
#include<cmath>
class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        for(int i=0;i<size(A);i++)
        {
            A[i]=pow(A[i],2);
        }
        sort(A.begin(),A.end());
        return A;
    }
};

方法二: 通过左右指针对应的数据的平方比较大小双向夹逼

时间复杂度:O(n)

空间复杂度O(n)

AC代码:

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        
        int len=A.size();
        int a=0,b=len-1,i=b;
        vector<int> B(len);
        while(i>=0)
        {
            //通过左右指针对应数据的平方比较大小,并向中间夹逼,避免找中间0的位置,同时降低了时间复杂度,妙!
            int left=A[a]*A[a],right=A[b]*A[b];
            if(left>right)
            {
                B[i]=left;
                a++;
            }
            else
            {
                B[i]=right;
                b--;
            }
            i--;
        }
        return B;
    }
};