leetcode 977.有序数组的平方 Java

题目链接

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

描述

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

提示：
1 <= A.length <= 10000
-10000 <= A[i] <= 10000
A 已按非递减顺序排序。

示例

示例 1：

输入：[-4,-1,0,3,10]
输出：[0,1,9,16,100]

示例 2：

输入：[-7,-3,2,3,11]
输出：[4,9,9,49,121]

初始代码模板

class Solution {
    public int[] sortedSquares(int[] A) {
    }
}

代码

困难题唯唯诺诺，简单题重拳出击，直接双指针

class Solution {
    public int[] sortedSquares(int[] A) {
       int[] ans = new int[A.length];

       int left = 0;
       int right = A.length - 1;

       for (int i = A.length - 1; i >= 0; i--) {
           if (A[left] * A[left] > A[right] * A[right]) {
               ans[i] = A[left] * A[left];
               left++;
           } else {
               ans[i] = A[right] * A[right];
               right--;
           }
       }
       
       return ans;
    }
}