有序数组的平方

71 阅读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) 的算法解决本问题

解题思路:首先可以确定的是这些数值的的平方最大的要么在最左边,要么在最右边;建一个跟原先数组一样长度的数组来存放结果,每次都把结果最大的放在新建数组的最右边,然后左右指针不断往中间挪

public class SortedSquares {
    public static void main(String[] args) {
        int nums[] = {-4,-1,0,3,10};
        int[] ints = sortedSquares(nums);
        for (int i = 0; i < ints.length; i++) {
            System.out.print(ints[i] +"-");
        }
    }
    public static 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;
    }
}