977.有序数组的平方（Java、暴力、双指针）

给你一个按 非递减顺序 排序的整数数组 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]

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

进阶：
请你设计时间复杂度为 O(n) 的算法解决本问题

LeetCode - 题目地址
代码随想录 - 题目详解

思路

这题如果使用暴力写法的话非常简单，只需要对数组元素进行平方处理之后进行排序。
针对数组、链表可以考虑是否能使用双指针法进行求解。
本题的前提条件有数组是有序的。返回结果也需要进行有序排列。
这时我们可以想到，在没有负数元素的情况下，数组元素平方处理之后还是有序的。
但是，这题是存在负数元素的。那么负数元素平方处理之后，很有可能由最小元素变为最大元素。
这时，对存在负数元素情况的的数组进行平方处理之后，数组元素中最大的元素就有可能分布在数组的两端。
可以使用双指针法， left 指针指向数组 nums 最左端位置， right 指针指向数组 nums最右端位置。
创建一个新数组 ans 作为排序之后的数组。ans 数组存在一个指针 k 指向 ans 数组最右端位置。
当 num[left] * nums[left] < nums[right] * nums[right] 时，ans[k] = nums[right] * nums[right]
当 num[left] * nums[left] > nums[right] * nums[right] 时，ans[k] = nums[left] * nums[left]

代码

暴力写法

class Solution {
    public int[] sortedSquares(int[] nums) {
        // 暴力写法：先对数组进行平方，后进行排序
        for (int i = 0; i < nums.length; i++) {
            nums[i] *= nums[i];
        }
        // 对数组进行排序
        Arrays.sort(nums);
        return nums;
    }
}

双指针法

class Solution {
    public int[] sortedSquares(int[] nums) {
        // 双指针法
        int left = 0;	// 左指针，指向数组最左端位置
        int right = nums.length - 1;	// 右指针，指向数组最右端位置
        int k = nums.length - 1;	// 指向新数组最右端位置
        // 保存排序之后的数组元素
        int[] ans = new int[nums.length];
		// 进行遍历，注意指针移动方向
        while ( left <= right ) {
            if (nums[left] * nums[left] < nums[right] * nums[right]) {
                ans[k--] = nums[right] * nums[right--];
            }else {
                ans[k--] = nums[left] * nums[left++];
            }
        }
        return ans;
    }
}