【双指针-简单】977. 有序数组的平方

菜菜2022

关注

原创 2022-09-18 16:15:12 189 阅读 AI 写同款· GEO 优化 ›

LeetCode977有序数组的平方（数组）算法特点及适用场景该算法采用双指针法，巧妙地利用了原数组有序的特点，高效地解决了有序数组元素平方后再排序的问题，避免了先对所有元素平方然后再进行常规排序（如使用快排等时间复杂度较高的排序算法）的复杂操作，时间复杂度仅为，代码实现简洁明了。适用于类似这种需要根据原数组的有序性以及特定元素关系（这里是元素平方大小比较）来构造新的有序数组的场景，只要满足可以通过双指针从两端向中间处理并根据一定规则填充新数组的情况，都可以考虑使用这种思路来解决问题。注意事项及可能遇到的问题。阅读详情

【题目】
在这里插入图片描述提示：

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序

进阶：

请你设计时间复杂度为 O(n) 的算法解决本问题

【代码】
在这里插入图片描述

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        right,left=len(nums),0
        for idx,item in enumerate(nums):
            if item>=0:                
                right=idx
                break
        rr=right
        left=right-1
        ans=[]
        while rr<len(nums) and left>=0:
            if nums[rr]<=abs(nums[left]):                
                ans.append(nums[rr]**2)
                rr+=1
            else:
                ans.append(nums[left]**2)
                left-=1
        while rr<len(nums):
            ans.append(nums[rr]**2)
            rr+=1
        while left>=0:
            ans.append(nums[left]**2)
            left-=1
        return ans

【方法2】
在这里插入图片描述

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

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        left,right=0,len(nums)-1
        ans=[0 for i in range(len(nums))]
        pos=len(nums)-1
        while left<=right:
            l=nums[left]**2
            r=nums[right]**2
            if l>=r:
                ans[pos]=l
                left+=1
            else:
                ans[pos]=r
                right-=1
            pos-=1
        return ans