其他分享
首页 > 其他分享> > LeetCode-1337. 矩阵中战斗力最弱的 K 行

LeetCode-1337. 矩阵中战斗力最弱的 K 行

作者:互联网

给你一个大小为 m * n 的矩阵 mat,矩阵由若干军人和平民组成,分别用 1 和 0 表示。

请你返回矩阵中战斗力最弱的 k 行的索引,按从最弱到最强排序。

如果第 i 行的军人数量少于第 j 行,或者两行军人数量相同但 i 小于 j,那么我们认为第 i 行的战斗力比第 j 行弱。

军人 总是 排在一行中的靠前位置,也就是说 1 总是出现在 0 之前。

执行用时:
24 ms, 在所有 Python3 提交中击败了100.00%的用户
内存消耗:
15 MB, 在所有 Python3 提交中击败了76.72%的用户
class Solution:
    def kWeakestRows(self, mat: List[List[int]], k: int) -> List[int]:
        num = {}
        l = []
        for i, j in enumerate(mat):
            num[i] = j.count(1)
        num = sorted(num.items(), key=lambda x: x[1])
        for i in range(k):
            l.append(num[i][0])
        return l

sorted相关排序知识可查看该文章

标签:num,mat,int,List,1337,矩阵,最弱,军人,LeetCode
来源: https://blog.csdn.net/qq_41542989/article/details/119294209