977. 有序数组的平方

88 阅读1分钟

给你一个按 非递减顺序 排序的整数数组 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) 的算法解决本问题

解题思路:新建一个与原数组一样长度的数组来接收数据,一个按 非递减顺序 排序的整数数组,每次取一个数字的平方最大的肯定是在左右两边,所以每次比较最左和最右两个数的平方比较大小,最大的依次从右到左存放到新数组中。

class Solution {
    public int[] sortedSquares(int[] nums) {
        int length = nums.length;
        int left = 0;
        int right = length -1;
        int[] newNums = new int[length];
        for (int i = length-1; i >= 0 ; i--){
            if( nums[right] *nums[right] > nums[left] *nums[left] ){
                newNums[i] = nums[right] *nums[right];
                right--;
            }else {
                newNums[i] = nums[left] *nums[left];
                left++;
            }
        }
        return newNums;
    }
}