2021-05-02

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

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

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

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

二分法+优先队列
用二分法求出每一行最后一个1的下标
创建一个类同时保存下标和对应行
用优先队列对其进行排序

class Solution {
    class pair{
        int power;
        int row;
        pair(int power, int row){
            this.power = power;
            this.row = row;
        }
    }
    public int[] kWeakestRows(int[][] mat, int k) {
        Queue<pair> minheap = new PriorityQueue<>(new Comparator<pair>(){
            public int compare(pair a, pair b){
                if(a.power != b.power)
                    return a.power - b.power;
                return a.row - b.row;
            }
        });
        for(int i = 0; i < mat.length; i++){
            minheap.offer(new pair(binarySearch(mat[i]), i));
        }
        int[] ans = new int[k];
        for(int i = 0; i < k; i++){
            ans[i] = minheap.poll().row;
        }
        return ans;
    }
    private int binarySearch(int[] nums){
        int l = 0, r = nums.length - 1;
        while(l + 1 < r){
            int mid = l + (r - l) / 2;
            if(nums[mid] == 0)
                r = mid;
            else if(nums[mid] == 1)
                l = mid;
        }
        if(nums[r] == 1)
            return r;
        if(nums[l] == 1)
            return l;
        return -1;
    }
}