#!/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%
的用户