【LeetCode刷题-Python】有序数组的平方

977.有序数组的平方

977.有序数组的平方

题目

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

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

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

代码

class Solution(object):
    def sortedSquares(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        n=len(nums)
        ans=[-1]*n #新建数组ans = [-1 for _ in range(n)]
        l,r,i=0,n-1,n-1

        while l<=r:
            if nums[l]*nums[l]>nums[r]*nums[r]:
                ans[i]=nums[l]*nums[l]
                l+=1
            else:
                ans[i]=nums[r]*nums[r]
                r-=1
            i-=1
        return ans

       

总结

• 复习双指针，向左右两边分别移动
• ans=[-1]*n #新建数组ans = [-1 for _ in range(n)]