leetcode 977. 有序数组的平方

leetcode 977. 有序数组的平方

一、题目

原题链接：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.解题思路

• 题目分析

1. 先遍历一次数组nums，将所有元素进行求平方

2. 数组首位使用双指针left和right左右夹击，将nums[left]和nums[right]进行比较，较大的push_back到新开辟的数组res中，最后得到的一定是按降序顺序的数组res。

3. 最后通过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;
      }
  };