【算法-数组】有序数组的平方

这里写自定义目录标题

• 一、题目
• 二、双指针解法

一、题目

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

示例 1：

输入：nums = [-4,-1,0,3,10]

输出：[0,1,9,16,100]

解释：平方后，数组变为 [16,1,0,9,100]

排序后，数组变为 [0,1,9,16,100]

二、双指针解法

不停地比较首尾元素平方的大小，并将较大的放到一个新的结果数组中

空间复杂度 o(n) 时间复杂度o(n)

public class SortArraySquare {

    public static int[] sortedSquares(int[] nums) {
        int[] res = new int[nums.length];
        int k = nums.length - 1;

        for (int i = 0, j = k; i <= j; ) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                res[k--] = nums[i] * nums[i];
                i++;
            } else {
                res[k--] = nums[j] * nums[j];
                j--;
            }
        }

        return res;
    }

    public static void main(String[] args) {

        int[] nums = {-4, -1, 0, 3, 10};
        int[] res = SortArraySquare.sortedSquares(nums);
        for (int i = 0; i < res.length; i++) {
            System.out.print(res[i] + " ");
        }
    }

}