【精选高频面试题】 Leetcode.977.有序数组的平方

Leetcode.977.有序数组的平方

https://leetcode-cn.com/problems/squares-of-a-sorted-array/

给你一个按 非递减顺序 排序的整数数组 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]

思路：这个题有两种解法：第一种暴力解法，先通过循环将数组元素全部平方，在调用库函数sort进行排序。第二种是通过双指针法解答，设置指向一头一尾的指针，再设置一个新数组，比较首尾指针，谁大，谁就被放入新数组的尾部。

解法一

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
           for(int i=0;i<nums.size();i++){
               nums[i]*=nums[i];
            }
        sort(nums.begin(),nums.end());
        return nums;
    }
};

解法二

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
          int k=nums.size()-1;
          vector<int>result(nums.size(),0);
          for(int i=0,j=k;i<=j;){
              if(nums[i]*nums[i]<nums[j]*nums[j]){
                  result[k--]=nums[j]*nums[j];
                  j--;
                }
              else (nums[i]*nums[i]>nums[j]*nums[j]){
                    result[k--]=nums[i]*nums[i];
                    i++;
              }
          }
          return result;
    }
};