解法一
- 插入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)]