997.有序数组的平方Java

997.有序数组的平方Java

题目描述

给你一个按非递减顺序排序的整数数组 nums，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。
尽量在时间复杂度为 O(N)内完成

输入输出样式

示例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,-1]
输出：[1,4,9,49]

本文题来自LeetCode：https://leetcode-cn.com/problems/squares-of-a-sorted-array

思路

前后两个指针同时进行扫描，比较所指两个元素平方大小，将较大者的平方逆序存在新构造的数组，直到指针相遇则退出循环。

算法分析

遍历了数组，时间复杂度为O(n),空间复杂度为O(1)

求解函数

public int[] sortedSquares(int[] nums) {
        int[] rst = new int[nums.length]; //新构造数组存结果数组
        int left = 0, right = nums.length - 1, i = nums.length - 1;
        while (left <= right){
            if (nums[left] * nums[left] >= nums[right] * nums[right]){
                rst[i--] = nums[left] * nums[left]; //逆序存储到rst
                left++;
            }
            else{
                rst[i--] = nums[right] * nums[right];
                right--;
            }
        }
        return rst;
    }

主函数调用验证结果

class Solution977 {
    public static void main(String[] args) {
        Solution977 sol = new Solution977();
        int[] nums = new int[]{-4,-1,0,3,10};
        int[] arr = sol.sortedSquares(nums);
        for (int i = 0; i < arr.length; i++) System.out.print(arr[i] + ", ");
    }
    public int[] sortedSquares(int[] nums){...}
}

输出结果

其他解法

直接遍历数组将所有元素平方，再调用排序函数排序，但时间复杂度大于O(n)，最快的快排也要O(nlogN)