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]

解题思路

先找出正负数的分界点，使用两个指针分别指向正数部分和负数部分，然后不断比较两个指针指向的元素，选出较小的那个元素的平方加入结果

代码

class Solution {
    public int[] sortedSquares(int[] nums) {

        int n=nums.length;
        int[] res=new int[n];
        int rp=0;int i=0;
        for(;i<n;i++)
            if(nums[i]>=0)
                break;
        int neg=i-1,pos=i;
        while(rp<n)
        {
            if(pos>=n||neg>=0&&Math.abs(nums[neg])<nums[pos])
            {
                res[rp]=nums[neg]*nums[neg];
                neg--;
            }else {
                res[rp]=nums[pos]*nums[pos];
                pos++;
            }
            rp++;
        }
        return res;
        
    }
}