題目描述

題幹：
給你一個大小為m* n的矩陣mat，矩陣由若干軍人和平民組成，分別用 1 和 0 表示。
請你返回矩陣中戰鬥力最弱的k行的索引，按從最弱到最強排序。
如果第i行的軍人數量少於第j行，或者兩行軍人數量相同但 i 小於 j，
那麼我們認為第 i 行的戰鬥力比第 j 行弱。
軍人 總是 排在一行中的靠前位置，也就是說 1 總是出現在 0 之前。

示例 1：
輸入：mat = 
[[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 = 3
輸出：[2,0,3]
解釋：
每行中的軍人數目：
行 0 -> 2 
行 1 -> 4 
行 2 -> 1 
行 3 -> 2 
行 4 -> 5 
從最弱到最強對這些行排序後得到 [2,0,3,1,4]

示例 2：
輸入：mat = 
[[1,0,0,0],
[1,1,1,1],
[1,0,0,0],
[1,0,0,0]], 
k = 2
輸出：[0,2]
解釋： 
每行中的軍人數目：
行 0 -> 1 
行 1 -> 4 
行 2 -> 1 
行 3 -> 1 
從最弱到最強對這些行排序後得到 [0,2,3,1]
 

題解思路

返回“戰鬥力”弱的k行行數，並且注意每一行的1都在0之前

每行1的數量就是“戰鬥力”，所以統計每行1的個數排序就可以

無論是陣列還是下面方法裡的最小堆都可以，如何統計1的個數也是迴圈和二分都能實現

正確程式碼

public int[] kWeakestRows(int[][] mat, int k) {
        int m = mat.length, n = mat[0].length;
        List<int[]> power = new ArrayList<>();
        // 二分找最後一個1下標
        for (int i = 0; i < m; i++) {
            int l = 0, r = n - 1, pos = -1;
            while (l <= r) {
                int mid = l + (r - l) / 2;
                if (mat[i][mid] == 0) {
                    r = mid - 1;
                } else {
                    pos = mid;
                    l = mid + 1;
                }
            }
            power.add(new int[]{pos + 1, i});
        }

        // 小頂堆
        PriorityQueue<int[]> priorityQueue = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if (o1[0] != o2[0]) {
                    return o1[0] - o2[0];
                } else {
                    return o1[1] - o2[1];
                }
            }
        });

        for (int[] ints : power) {
            priorityQueue.offer(ints);
        }

        // 只取前k行
        int[] ans = new int[k];
        for (int i = 0; i < k; i++) {
            ans[i] = priorityQueue.poll()[1];
        }
        return ans;
    }
 

總結

堆的offer和add方法與最小堆的比較方法值得注意，還有就是八月一建軍節快樂！

如果文章存在問題或者更好的題解，歡迎在評論區斧正和評論，各自努力，你我最高處見