Day02 数组理论基础 | 977.有序数组的平方，209.长度最小的子数组，59.螺旋矩阵II

最新推荐文章于 2024-08-13 18:29:56 发布

派派派巨星

最新推荐文章于 2024-08-13 18:29:56 发布

阅读量200

点赞数 3

文章标签： python 算法数据结构 leetcode

本文链接：https://blog.csdn.net/P_YOU/article/details/136582641

版权

LeetCode 977 有序数组的平方

题目描述：给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。
输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

最直接的方法就是每个数平方之后，再整体重新排序，但复杂度较高。

双指针法

思路

靠近左右两边的数，平方后一般会更大
定义两个指针，初始指向头和尾位置
两个指针逐步往中间靠拢，并比较左右指针所指的元素平方的大小
大的元素平方填入新数组，对应指针往中间靠拢一位

代码

def sortedSquares(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        left, right = 0, len(nums) - 1
        result = [0 for _ in range(len(nums))]
        k = len(nums) - 1
        while left <= right:
            if nums[left] * nums[left] > nums[right] * nums[right]:
                result[k] = nums[left] * nums[left]
                left += 1
            else:
                result[k] = nums[right] * nums[right]
                right -= 1
            k -= 1
        return result