力扣(C++): 有序数组的平方

题目描述

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

示例 1：

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

示例 2：

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

题目来源力扣网.

代码示例:

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

代码较简单, 平方排序即可