leetcode977. 有序数组的平方

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