给你一个按 非递减顺序 排序的整数数组 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]
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序
双指针法:谁的绝对值大,计算平方后谁先insert到vector的头
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> ret;
int left = 0;
int right = nums.size() - 1;
while(left <= right) {
nums[left] = abs(nums[left]);
nums[right] = abs(nums[right]);
if(left == right){
ret.insert(ret.begin(), nums[left] * nums[left]);
break; // left = right时,跳出循环
}
if(nums[left] >= nums[right]){
ret.insert(ret.begin(),nums[left] * nums[left]);
left++;
}else{
ret.insert(ret.begin(),nums[right] * nums[right]);
right--;
}
}
return ret;
}
};
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。