Leetcode【977】---Squares of a Sorted Array排序数组的平方【数组】【Easy】

题目描述

给定一个整数数组A，按非降序存储，返回数组内每个数的平方，也是按非降序存储的。

样例1

Input: [-4,-1,0,3,10]
Output: [0,1,9,16,100]

样例2

Input: [-7,-3,2,3,11]
Output: [4,9,9,49,121]

注意

• 1 <= A.length <= 10000
• -10000 <= A[i] <= 10000
• A是按非降序存储的

思路

对数组元素求平方后，直接进行插入排序

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        
        for(int i=0;i<A.size();i++){
            A[i]=A[i]*A[i];
            int key=A[i];
            int j=i-1;
            
            while(j>=0&&A[j]>key){
                A[j+1]=A[j];
                j--;
            }
            A[j+1]=key;
        }
        return A;
        
    }
};

测试