有序数组的平方python3(leetcode977)

#977. 有序数组的平方

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

输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]

#sorted方法
class Solution(object):
    def sortedSquares(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        for i in range(0, len(nums)):
            nums[i] = nums[i]*nums[i]
        return sorted(nums)
##双指针法
class Solution(object):
    def sortedSquares(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        #双指针法
        n = len(nums)
        l, r, k = 0, n-1, n-1
        ans = [0]* n
        while l <= r:
            l_value, r_value = nums[l]* nums[l], nums[r]* nums[r]

            if l_value < r_value:
                ans[k] = r_value
                r -=1
            else:
                ans[k] = l_value
                l +=1
            k -=1         
        return ans

猜你喜欢

转载自blog.csdn.net/ziqingnian/article/details/121795287