Python描述LeetCode1337.矩阵中战斗力最弱的K行

  大家好&＃xff0c;我是亓官劼&＃xff08;qí guān jié &＃xff09;&＃xff0c;在【亓官劼】公众号、CSDN、GitHub、B站等平台分享一些技术博文&＃xff0c;主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文&＃xff0c;时光荏苒&＃xff0c;未来可期&＃xff0c;加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)&＃xff0c;里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言&＃xff0c;我会尽快回复消息.

---

本文原创为【亓官劼】&＃xff08;qí guān jié &＃xff09;&＃xff0c;请大家支持原创&＃xff0c;部分平台一直在恶意盗取博主的文章&＃xff01;&＃xff01;&＃xff01; 全部文章请关注微信公众号【亓官劼】。

题目

给你一个大小为 m * n 的矩阵 mat&＃xff0c;矩阵由若干军人和平民组成&＃xff0c;分别用 1 和 0 表示。

请你返回矩阵中战斗力最弱的 k 行的索引&＃xff0c;按从最弱到最强排序。

如果第 i 行的军人数量少于第 j 行&＃xff0c;或者两行军人数量相同但 i 小于 j&＃xff0c;那么我们认为第 i 行的战斗力比第 j 行弱。

军人 总是 排在一行中的靠前位置&＃xff0c;也就是说 1 总是出现在 0 之前。

示例 1&＃xff1a;

输入&＃xff1a;mat &＃61;
[[1,1,0,0,0],[1,1,1,1,0],[1,0,0,0,0],[1,1,0,0,0],[1,1,1,1,1]],
k &＃61; 3
输出&＃xff1a;[2,0,3]
解释&＃xff1a;
每行中的军人数目&＃xff1a;
行 0 -> 2
行 1 -> 4
行 2 -> 1
行 3 -> 2
行 4 -> 5
从最弱到最强对这些行排序后得到 [2,0,3,1,4]


示例 2&＃xff1a;

输入&＃xff1a;mat &＃61;
[[1,0,0,0],[1,1,1,1],[1,0,0,0],[1,0,0,0]],
k &＃61; 2
输出&＃xff1a;[0,2]
解释&＃xff1a;
每行中的军人数目&＃xff1a;
行 0 -> 1
行 1 -> 4
行 2 -> 1
行 3 -> 1
从最弱到最强对这些行排序后得到 [0,2,3,1]


提示&＃xff1a;

• m &＃61;&＃61; mat.length
• n &＃61;&＃61; mat[i].length
• 2 <&＃61; n, m <&＃61; 100
• 1 <&＃61; k <&＃61; m
• matrix[i][j] 不是 0 就是 1

Python描述一

排序后输出前k即可

class Solution:def kWeakestRows(self, mat: List[List[int]], k: int) -> List[int]:sum_list &＃61; [{"idx":idx,"sum":sum(_)} for idx,_ in enumerate(mat)]sum_list.sort(key&＃61;lambda item:(item[&＃39;sum&＃39;],item[&＃39;idx&＃39;]))return [sum_list[idx][&＃39;idx&＃39;] for idx in range(k)]