LeetCode - 704 - Binary Search

2023-05-17 14:46:48 浏览数 (1)

# Squares Of A Sorted Array

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

示例 1:

代码语言:javascript复制
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]

示例 2:

代码语言:javascript复制
输入:nums = [-7,-3,2,3,11]
输出:[4,9,9,49,121]

提示:

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

# 直接排序

# 思路

将数组元素全部平方后直接排序

# 实现

代码语言:javascript复制
/**
 * @param {number[]} nums
 * @return {number[]}
 */
var sortedSquares = function(nums) {
    const arr = nums.map(num => num ** 2);
    return arr.sort((a, b) => (a - b));
};

# 复杂度

  • 时间:O(n log n),n 是数组 nums 的长度
  • 空间:O(log n),除了存储结果数组外,需要 O(log n) 的栈空间进行排序

# 双指针

# 思路

直接排序的话,其实没有利用到 数组 nums 已经按照升序排序 这个条件。显然,如果数组 nums 中的所有数都是非负数,那么将每个数平方之后,数组仍保持升序;如果数组 nums 中的所有数都是负数,那么将每个数平方之后,数组会保持降序。

这样一来,如果能够找到数组 nums 中的负数与非负数的分界线,那么就可以用类似 归并排序 (opens new window) 的方法。可以设 neg 为数组 nums 中的负数与非负数的分界线,即,nums[0]nums[neg] 均为负数,而 nums[neg 1]nums[nums.length - 1] 均为非负数。进行平方处理后,有 nums[0]nums[neg] 单调递减,而 nums[neg 1]nums[nums.length - 1] 单调递增。

在得到两个有序的子数组后,就可以使用归并的方法进行排序了。使用两个指针分别指向位置 negneg 1,每次比较两个指针对应的数,选择较小的那个放入结果数组,并移动指针。当某一指针移至边界时,将另一指针还未遍历到的数全部依次放入结果数组即可。

还可以优化一步,不用考虑边界情况:使用两个指针分别指向位置 0n - 1, 选择比较大的,逆序 放入结果数组,并移动指针。

# 实现

代码语言:javascript复制
/**
 * @param {number[]} nums
 * @return {number[]}
 */
var sortedSquares = function(nums) {
    const len = nums.length;
    let left = 0;
    let right = len - 1;
    let result = [];
    while (left <= right) {
        let lval = nums[left];
        let rval = nums[right];
        if (Math.abs(lval) >= Math.abs(rval)) {
            result.unshift(lval ** 2);
            left  ;
        } else {
            result.unshift(rval ** 2);
            right--;
        }
    }
    return result;
};

# 复杂度

  • 时间:O(n),n 是数组 nums 的长度
  • 空间:O(1)

0 人点赞