有序数组的平方（LeetCode）

好好学习Py

于 2024-08-08 22:20:59 发布

阅读量208

点赞数 4

分类专栏：算法与数据结构文章标签： leetcode 数据结构算法 python

本文链接：https://blog.csdn.net/weixin_74254879/article/details/141037444

版权

算法与数据结构专栏收录该内容

32 篇文章 0 订阅

订阅专栏

题目

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

解题

以下算法时间复杂度为 $O(n)$

def sortedSquares(nums):
    n = len(nums)
    result = [0] * n  # 创建一个结果数组，长度与 nums 相同
    left, right = 0, n - 1  # 初始化左右指针
    position = n - 1  # 初始化结果数组的插入位置

    while left <= right:
        left_square = nums[left] ** 2
        right_square = nums[right] ** 2
        if left_square > right_square:
            result[position] = left_square
            left += 1
        else:
            result[position] = right_square
            right -= 1
        position -= 1

    return result


nums = [-4, -1, 0, 3, 10]
print(sortedSquares(nums))  # 输出: [0, 1, 9, 16, 100]

nums = [-7, -3, 2, 3, 11]
print(sortedSquares(nums))  # 输出: [4, 9, 9, 49, 121]