2024.3.7力扣(<=1200)刷题记录-977. 有序数组的平方

一、题目

977. 有序数组的平方

二、解题

1.运用内置函数

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        #内置函数
        n = len(nums)
        ans = [0] * n
        for i in range(n):
            ans[i] = nums[i] * nums[i]
        ans.sort()    # 时复nlogn
        return ans

 先将每一位平方,再排序。但是sort()函数的时间复杂度为O(nlogn),不满足题目要求。

2.相向双指针

为什么会想到使用这个方法呢?因为题目有一个很重要的点就是nums数组是非递减排序的,也就是不严格的递增。对于有顺序的数组,一定要充分利用有序性。

下面对题目进行分析:

在0两侧的尽头的绝对值最大,越靠近中间的0越小,那么将两个指针指向左右尽头,平方大的就先放入数组,然后移动指针。而平方大实际上就是绝对值大,而绝对值大就是“-负数”和“+正数”比较谁更大。而L指针指向的是负数,R指针指向的是正数,所以判断条件为- nums[l] >= nums[r]。

而值为0的时候是一个特殊的存在,当指针所指的值为0时,就不需要再比较两指针的绝对值大小了。因为剩下的数值里面是符号相同的数,从指向0的指针到另一个指针的值的绝对值必然是增加的(因为nums数组是递增的,充分利用有序性)。如若剩下的数组为【-4,-3,-2,-1,0】,此时即L指向-4,R指向0,那么从0到-4的绝对值必然是增加的。然后我们将这段数组的值直接存入答案数组即可。注意这里要从远离0的位置开始存,因为我们之前一直是将当前的绝对值大值存入数组,这里也是一样。

另外,博主在写代码的时候,先是将ans数组(答案数组)令为空,然后使用insert()函数在列表的头位(下标为0的位)插入的数值(要求非递减顺序),但是运行起来发现时间太慢了,所以改为了idx存需要放的下标,再赋值。原代码为注释的代码。

代码如下:

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        # 双指针
        n = len(nums)
        # ans = []
        ans = [0] * n
        idx = n-1
        l,r = 0,n-1     #指针
        while l <= r:
            if nums[l] == 0:
                for i in range(r-l+1):
                    # ans.insert(0,nums[r-i] ** 2)
                    ans[idx] = nums[r-i] ** 2
                    idx -= 1
                break
            if nums[r] == 0:
                for i in range(r-l+1):
                    # ans.insert(0,nums[l+i] ** 2)
                    ans[idx] = nums[l+i] ** 2
                    idx -= 1
                break
            if - nums[l] >= nums[r]:
                # ans.insert(0,nums[l] ** 2)
                ans[idx] = nums[l] ** 2
                idx -= 1
                l += 1
            else:
                # ans.insert(0,nums[r] ** 2)
                ans[idx] = nums[r] ** 2
                idx -= 1
                r -= 1
        return ans

感谢你看到这里!一起加油吧!

  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值