1. 题目信息
给定一个按非递减顺序排序的整数数组 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 已按非递减顺序排序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 找到数组中第一个非负数下标 i,第一个负数下标 j = i - 1;
- 从中间向两边比较 abs(A[ j ]) 和 A[ i ],取小的数的平方push进入答案数组ans
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int i, j;
for (i = 0; i < A.size() && A[i] < 0; ++i)
{
//find the index of nonpositive value
}
j = i-1;
vector<int> ans;
while(j >= 0 && i < A.size())
{
if(A[i] < abs(A[j]))
{
ans.push_back(A[i]*A[i]);
++i;
}
else
{
ans.push_back(A[j]*A[j]);
--j;
}
}
if(j < 0)
{
while(i < A.size())
{
ans.push_back(A[i]*A[i]);
++i;
}
}
else//i到达末尾
{
while(j >= 0)
{
ans.push_back(A[j]*A[j]);
--j;
}
}
return ans;
}
};