题意:给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
思路:拿到题目我们最直接的思路是先遍历一遍平方,然后在排序,当然可以解决问题,复杂度为O(NlogN);
我们这里主要介绍一种O(N)的算法,因为原序列是有有序的,那么平方以后的最大值只会出现在两端,所以我们可以采用双指针的方法,从两头开始比较,看哪端大。
- 假设右端平方大于左端,则确定一个最大值,然后右端指针往左移。
- 若左端平方大于右端,则也确定一个最大值,然后左端指针往右移。
如此遍历一遍,即可。
code
class Solution {
public:
vector<int> sortedSquares(vector<int>& A) {
int k = A.size() - 1;
vector<int> result(A.size(), 0);
for (int i = 0, j = A.size() - 1; i <= j;) { // 注意这里要i <= j,因为最后要处理两个元素
if (A[i] * A[i] < A[j] * A[j]) {
result[k--] = A[j] * A[j];
j--;
}
else {
result[k--] = A[i] * A[i];
i++;
}
}
return result;
}
};