Java - 977. 有序数组的平方 - 进阶O(n)复杂度

一、题目

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

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

二、思路

1 先把原数组求平方

2 将平方后的数组理解为2个有序数组，一个是数组头到最小值的降序，一个是数组最小值到数组尾的升序

3 原题目变为2个有数素组合并，用low，high2个指针依次从原数组两头向中间移动比较，较大的赋值给结果数组的当前尾部

4 具体思路和2个有序数组/2个有序链表的合并完全一致

三、代码

class Solution {
    public int[] sortedSquares(int[] nums) {
        if (nums.length == 1) {
            nums[0] = nums[0] * nums[0];
            return nums;
        }
        for (int i = 0; i < nums.length; i++) { //遍历求平方
            nums[i] = nums[i] * nums[i];
        }
        int low = 0;
        int high = nums.length - 1;
        int temp = nums.length - 1;
        int[] result = new int[nums.length];
        while (low < high) {  //双指针头尾依次对比赋值
            if (nums[low] < nums[high]) {
                result[temp] = nums[high];
                high--;
            } else {
                result[temp] = nums[low];
                low++;
            }
            temp--;
        }
        result[temp] = nums[low]; //还有low==high的位置需要赋值
        return result;
    }

}