leetcode 977. 有序数组的平方
一、题目
1.题目描述
给你一个按 非递减顺序 排序的整数数组 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 <= 10^4
-10^4 <= nums[i] <= 10^4
nums
已按 非递减顺序 排序
2.基础框架
C++基础框架代码如下:
vector<int> sortedSquares(vector<int>& nums) {
}
3.解题思路
- 题目分析
-
先遍历一次数组
nums
,将所有元素进行求平方 -
数组首位使用双指针
left
和right
左右夹击,将nums[left]
和nums[right]
进行比较,较大的push_back到新开辟的数组res
中,最后得到的一定是按降序顺序的数组res
。 -
最后通过
reverse()
操作,得到升序数组并进行返回。
-
复杂度分析:
时间复杂度:O(n)
空间复杂度:O(n)
-
实现代码:
class Solution { public: vector<int> sortedSquares(vector<int>& nums) { int n = nums.size(); vector<int> res; res.reserve(n); // 优化: 减少push_back操作的开销 for (int i = 0; i < n; ++i) nums[i] = pow(nums[i], 2); int left = 0, right = n - 1; while (left <= right) { if (nums[left] >= nums[right]) { res.push_back(nums[left]); ++left; } else { res.push_back(nums[right]); --right; } } reverse(res.begin(), res.end()); return res; } };