给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
示例 2:
输入:nums = [-7,-3,2,3,11]
输出:[4,9,9,49,121]
题解
数组是有序的,但是存在负数,所以无法判断平方后的大小。可以设置 2 个指针,一前一后,对比绝对值的大小,再逐步向中心靠拢
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> res(nums.size(), 0);
int i = 0;
int j = nums.size() - 1;
for (int n = nums.size() - 1; n >= 0; n--) {
if (abs(nums[j]) > abs(nums[i])) {
res[n] = nums[j] * nums[j];
j--;
} else {
res[n] = nums[i] * nums[i];
i++;
}
}
return res;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。