1337. 方阵中战斗力最弱的 K 行
2020-02-12 本文已影响0人
鱼欲遇雨
解法一
- 插入index和对应关系的值的时候,使用sum的值+i的值,这个时候如果sum相同,i值小的排在前面
代码
class Solution:
def kWeakestRows(self, mat: List[List[int]], k: int) -> List[int]:
n = len(mat)
m = len(mat[0])
x = [[sum(l)*(n+1)+(i), i] for i,l in enumerate(mat)]
x.sort(key=lambda x: x[0])
return [x[i][1] for i in range(k)]