有序数组的平方

首先我们使用暴力解法,直接遍历列表的每个元素平方后,在进行排序输出。但是时间复杂度 为O(nlogn)。

代码如下:

#普通暴力法
class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        a = []
        for i in nums:
            a.append(i * i)
        return sorted(a)
#暴力法+列表推导式
class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        return sorted(x*x for x in nums)

这道题有说明让我们设计时间复杂度为O(n)的算法,所以这里我们使用双指针

双指针法

数组其实是有序的, 只不过负数平方之后可能成为最大数了。

那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。

此时可以考虑双指针法了,i指向起始位置,j指向终止位置。

定义一个新数组result,和A数组一样的大小,让k指向result数组终止位置。

如果A[i] * A[i] < A[j] * A[j] 那么result[k--] = A[j] * A[j]; 。

如果A[i] * A[i] >= A[j] * A[j] 那么result[k--] = A[i] * A[i]; 。

n = list(map(int, input().split(",")))
i = 0
j = len(n) - 1
k = len(n) - 1
result = [0] * len(n)  # 提前定义一个和 n 长度相同的列表,初始化为0
while i <= j:
    if n[i] ** 2 < n[j] ** 2:
        result[k] = n[j] ** 2
        j -= 1
    else:
        result[k] = n[i] ** 2
        i += 1
    k -= 1
print(result)

时间复杂度为O(n),其中这里用的是双指针的办法,这里初始化的指针有三个,i和j是在输入的数组上进行操作,而k是在最终输出的数组进行操作,其中k刚开始就在数组的末尾,然后经过判断后,最终的result数组是按照从小到大的顺序输出。

  • 12
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值