LeetCode 977 - 有序数组的平方 - C++含代码

题目（LeetCode 977）
给定一个按非递减顺序排序的整数数组 A，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。
(1) 1 <= A.length() <= 10000
(2) -10000 <= A[i] <= 10000
(3) A 已按非递减顺序排序。

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

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

---

方法一：排序

一种简单的思路是依次求出A中所有数字的平方，然后对该数组按非递减顺序排序。代码如下。

vector<int> sortedSquares(vector<int>& A) {
   
    vector<int> ans(A.size(), 0);
    for (int i = 0; i < A.size(); i++)
        ans[i] = A[i] * A[i];
    sort(ans.begin(), ans.end());
    return ans;
}

我们对整个数组遍历了一次，时间复杂度是$O(n)$，使用sort()函数排序的时间复杂度是 O ( n log ⁡ n ) O(n\log n)