Leetcode 977. 有序数组的平方

977. 有序数组的平方

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

进阶：

请你设计时间复杂度为 O(n) 的算法解决本问题

思路：
找到有序数组中负数与非负数的分界，使用类似于归并排序的方法将之排序后放入新数组。
代码如下：

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
            vector<int> ans;
            int zer=nums.size();
            for(int i=0;i<nums.size();i++){
                if(nums[i]>=0)
                {
                    zer=i;
                    break;
                }

            }
            
            //0->zer <0      zer->size()-1>0
            int l=zer-1;
            int r=zer;
            // cout<<l<<" "<<r;
            while(true){
                if(l>=0&&r<nums.size()&&0-nums[l]<nums[r]){
                    ans.push_back(nums[l]*nums[l]);
                    l--;
                }
                else if(l>=0&&r<nums.size()&&0-nums[l]>=nums[r]){
                    ans.push_back(nums[r]*nums[r]);
                    r++;
                }
                else if(l>=0&&r==nums.size()){
                    ans.push_back(nums[l]*nums[l]);
                    l--;
                }
                else if(l==-1&&r<nums.size()){
                    ans.push_back(nums[r]*nums[r]);
                    r++;
                }
                else if(l==-1&&r==nums.size()){
                    break;
                }
            }
            return ans;
        
            

    }
};