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

原创已于 2022-09-18 16:15:12 修改 · 174 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

于 2022-09-18 16:07:50 首次发布

刷题专栏收录该内容

643 篇文章

订阅专栏

本文介绍了一种针对已排序数组，将其每个元素平方后再次排序的算法实现。提供了两种解决方案，均采用双指针技术，一种使用插入排序方式构建结果数组，另一种直接定位元素位置填充结果数组。这两种方法的时间复杂度均为O(n)，满足了高效处理的要求。

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

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