https://leetcode.com/problems/the-k-weakest-rows-in-a-matrix/description
1337. The K Weakest Rows in a Matrix
給你一個陣列mat[][],mat[i][j] = 0 表示市民,1 表示士兵,給予一個數字 k
,如果一個列的士兵越少這個列的守備越薄弱,如果士兵一樣多比較上面的列更
薄弱,求出前 k 個守備薄弱的列。
思路:
1.用一個 Heap 儲存每一列的士兵數量和列編號,依照題目要求排序。
2.從 Heap 取出 k 個元素,把他們的編號返回即可。
Java Code: