力扣刷题打卡day9 有序数组的平方977

题目描述

给你一个按 非递减顺序排序的整数数组 nums，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

思路一：先平方再排序

• 现将整个数组元素平方，再用快排升序排序
• 时间复杂度为$O(n\log n)$

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

思路二：从零点开始向两侧扩散

• 先设两个变量pt1，pt2，将它们移到零点两端将元素依次从小至大放进新容器内，最后返回新创建的vector
• 空间复杂度为$O(n)$
• 时间复杂度为$O(n)$

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
       int pt1=0,pt2=0;
       vector<int>res;
       while(true){
            if(nums[pt1]>=0){
                break;
            }
           if(pt1<nums.size()){
               if(pt1+1==nums.size()){
                   break;
               }
               pt1++;
           }else{
               for(int i=0;i<nums.size();i++){
                   res.push_back(nums[i]);
               }
               return res;
           }
       }
       pt2=pt1;
       pt1--;
       while(pt1>=0||pt2<nums.size()){
           if(pt1>=0&&pt2>=nums.size()){
               while(pt1>=0){
                   res.push_back(pow(nums[pt1],2));
                   pt1--;
               }
               break;
           }else if(pt1<0&&pt2<=nums.size()){
               while(pt2<nums.size()){
                   res.push_back(pow(nums[pt2],2));
                   pt2++;
               }
               break;
           }else{
                if(nums[pt2]>=abs(nums[pt1])){
                    res.push_back(pow(nums[pt1],2));
                    pt1--;
                }else{
                    res.push_back(pow(nums[pt2],2));
                    pt2++;
                }
           }
       }
       return res;
    }
};