C++刷題-有序數組的平方

題目

給定一個按非遞減順序排序的整數數組 A，返回每個數字的平方組成的新數組，要求也按非遞減順序排序。

示例

輸入：[-4,-1,0,3,10]
輸出：[0,1,9,16,100]
輸入：[-7,-3,2,3,11]
輸出：[4,9,9,49,121]

代碼

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