文巾解题 977. 有序数组的平方

Posted 2021-09-01 刘文巾

 

 2 解题思路

2.1 直接平方然后排序

class Solution(object):
    def sortedSquares(self, nums):
        return sorted([x*x for x in nums])

2.2 双指针

因为原来的数组是非递减顺序，所以我们让双指针一个指向数组的首位置，一个指向数组的末位置。如果哪个绝对值大，那么将哪个的平方放到我们的数组中去，最后，将数据倒序即可

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