1337. 方阵中战斗力最弱的 K 行

2020-02-12  本文已影响0人  鱼欲遇雨

解法一

  1. 插入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)]
上一篇下一篇

猜你喜欢

热点阅读