leetcode977. 有序数组的平方

lanlinbuaa

于 2024-01-18 00:29:13 发布

阅读量366

点赞数 6

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.csdn.net/lanlinjnc/article/details/135663362

本文介绍了如何使用双指针法解决LeetCode中一个题目，即给定一个整数数组，找到数组平方后排序的最大值。通过比较数组两端元素的平方，逐步缩小搜索范围，最后返回排序后的平方数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode链接

题目

在这里插入图片描述

思路

最直观暴力的想法，莫过于每个数平方之后，排个序

数组其实是有序的，只不过负数平方之后可能成为最大数了。那么数组平方的最大值就在数组的两端，不是最左边就是最右边，不可能是中间。

此时可以考虑双指针法了，i指向起始位置，j指向终止位置。

代码

代码1 双指针法（头尾指针）

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        left = 0
        right = len(nums)-1
        result = []
        while left<=right:
            if nums[left]**2<nums[right]**2:
                result.append(nums[right]**2)
                right -= 1
            else:
                result.append(nums[left]**2)
                left += 1
        return result[::-1]

代码2 暴力破解

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        for i in range(len(nums)):
            nums[i] *= nums[i]
        nums.sort()
        return nums