LeetCode题解之有序数组的平方

除数博弈

题目描述

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

示例 1：

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

示例 2：

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

提示：

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

解题思路

创建一个数组，然后将元数组 A 的平方存储在该数组中，然后对该数组进行排序即可。

复杂度分析

• 时间复杂度：O(nlogn)，其中 n 为数组 A 的长度。
• 空间复杂度：O(1)，除了输出的额外空间复杂度为 O(1)。

代码实现

class Solution {
    public int[] sortedSquares(int[] A) {
        int[] arr = new int[A.length];
        for (int i = 0; i < A.length; i ++) {
            arr[i] = A[i] * A[i];
        }
        Arrays.sort(arr);
        return arr;
    }
}