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

菜菜2022

已于 2022-09-18 16:15:12 修改

阅读量134

点赞数

分类专栏：刷题文章标签： leetcode 算法职场和发展

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

本文链接：https://blog.csdn.net/kz_java/article/details/126918913

版权

刷题专栏收录该内容

643 篇文章 5 订阅

订阅专栏

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

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