977.有序数组的平方

题目链接

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

示例1:

  1. 输入:nums = [-4,-1,0,3,10]
  2. 输出:[0,1,9,16,100]
  3. 解释:平方后,数组变为 [16,1,0,9,100]
  4. 排序后,数组变为 [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),没有要求原地

first指针指向数组第一个元素,last指针指向数组最后一个元素,选其中大的

class Solution {
    public int[] sortedSquares(int[] nums) {
        int length = nums.length;
        int[] result = new int[length];
        int first = 0;
        int last = length - 1;
        int i = length - 1;
        while (first <= last) {
            if(nums[first]*nums[first] > nums[last]*nums[last]){
                result[i] = nums[first]*nums[first];
                first++;
            }else {
                result[i] = nums[last]*nums[last];
                last--;
            }
            i--;
        }
        return result;
    }
}