1337. 矩阵中战斗力最弱的k行

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# author:wu_tianDe
# datetime:2021/8/1 20:57
# software: PyCharm

class Solution:
    def kWeakestRows(self, mat, k: int):
        jun = {}
        for i,item in enumerate(mat): # 计算每行的军人数值
            jun[i] = sum(item)

        resD = sorted(jun.items(),key=lambda kv:(kv[1],kv[0])) # 字典按值排序
        # print(resD)
        res = []
        for i in range(k): # 将前k行的键的结果保存
            res.append(resD[i][0])

        return res # 返回结果


if __name__ == '__main__':
    s = Solution()
    mat = [[1,0,0,0],
 [1,1,1,1],
 [1,0,0,0],
 [1,0,0,0]]
    k = 2
    print(s.kWeakestRows(mat,k))

执行结果：
通过
显示详情
添加备注

执行用时：
32 ms
, 在所有 Python3 提交中击败了
95.41%
的用户
内存消耗：
15 MB
, 在所有 Python3 提交中击败了
81.31%
的用户