给定一个按非递减顺序排序的整数数组 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
已按非递减顺序排序。
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int N=A.size();
vector<int> res(N);
if(A[0]>=0)
{
for(int i=0;i<N;i++)
res[i]=A[i]*A[i];
return res;
}
int i=N-1,j=0,t=N-1;
while(i>=j)
{
int ai=A[i]*A[i];
int aj=A[j]*A[j];
if(ai<aj)
{
res[t--]=aj;
j++;
}
else
{
res[t--]=ai;
i--;
}
}
return res;
}
};