python中的sort()是原地操作,没有返回值

这篇博客介绍了如何在一个由军人和平民组成的矩阵中找到战斗力最弱的k行,并按顺序返回其索引。作者通过定义一个自定义的比较函数实现了这一目标,特别强调了Python中sort()和sorted()的区别。文章以一个具体的例子展示了算法的实现,并给出了正确的输出结果。此外,还提醒读者注意sort()原地操作的特性。
摘要由CSDN通过智能技术生成

lc1337/
给你一个大小为 m * n 的矩阵 mat,矩阵由若干军人和平民组成,分别用 1 和 0 表示。 请你返回矩阵中战斗力最弱的 k
行的索引,按从最弱到最强排序。如果第 i 行的军人数量少于第 j 行,或者两行军人数量相同但 i 小于 j,那么我们认为第 i
行的战斗力比第 j 行弱。军人 总是 排在一行中的靠前1.0输入:

mat = 
[[1,1,0,0,0],
 [1,1,1,1,0],
 [1,0,0,0,0],
 [1,1,0,0,0],
 [1,1,1,1,1]], 
k = 3

输出:[2,0,3]
class Solution:
    def kWeakestRows(self, mat: list[list[int]], k: int) -> list[int]:
        # 可以根据每一行的大小将她们的索引进行从大到小排序。输出前k个索引。
        index_list = sorted(list(range(len(mat))),key=lambda x :self.keyAccording( index = x,mat=mat))
        # 特别注意下面这种错误的写法,sort()是原地操作,没有返回值。
        #index_list = list(range(len(mat))).sort(key=lambda x :self.keyAccording( index = x,mat=mat))  
        return index_list[:k]

    def keyAccording(self,  mat: list[list[int]], index):
        left = 0
        right = len(mat[index]) - 1
        ans = None
        while left <= right:
            mid = (left + right) // 2
            if mat[index][mid] == 1:
                left = mid + 1
                ans = mid
            else:
                right = mid - 1
        if ans == None:
            return 0
        else:
            return ans + 1 

mat = [[1,1,0,0,0],
 [1,1,1,1,0],
 [1,0,0,0,0],
 [1,1,0,0,0],
 [1,1,1,1,1]]


k = 3

aa = Solution()
print(aa.kWeakestRows(mat,k))

输出:

[2, 0, 3]

本文的创新点是自定义了比较函数,头一回应用,记录一下。
需要注意的就是:
sort()是进行原地操作,没有返回值。
sorted()是非原地操作,不影响原列表的排序,有返回值。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值