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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

示例 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) 的算法解决本问题

思路

  • 根据题目可以总结出,数组是递增的顺序,因为说 是 非递减并且是排序的数组
  • 那么平方后,最大的数肯定是两头的数字,这里想到用双指针进行对比两头数字
  • 每次比较,大的值则插入返回数组中的头部
/**
 * @param {number[]} nums
 * @return {number[]}
 */
var sortedSquares = function (nums) {
    let left = 0,
        right = nums.length - 1,
        result = []
    while (left <= right) {
        if (left == right) { result.unshift(Math.pow(nums[right], 2)); left++ } //边界处理
        else {
            if (Math.abs(nums[left]) < Math.abs(nums[right])) { //右侧大,则加入数组  右指针减1
                result.unshift(Math.pow(nums[right], 2))
                right--
            } else {
                result.unshift(Math.pow(nums[left], 2)) //反之
                left++
            }
        }
    }
    return result;
};
Last modification:February 19, 2023
如果觉得我的文章对你有用,请随意赞赏