力扣 1337. 矩阵中战斗力最弱的 K 行

题目来源：https://leetcode-cn.com/problems/the-k-weakest-rows-in-a-matrix/

大致题意：
给出一个由0和1组成的矩阵，返回k个行内总数较小的行索引。
注意，所有的1（如果有的话）都会在0之前。

思路

1. 遍历矩阵所有行
2. 对于每一行，二分查找最后一个1的位置，该位置加 1 就是该行的数值和
3. 遍历时，存下索引和对应的数值和
4. 按照数值和排序，相同数值和行数大的行更大

代码：

public int[] kWeakestRows(int[][]  mat, int k) {
        List<int[]> rowList = new ArrayList<int[]>();
        int index = 0;
        int n = mat[0].length;
        // 遍历所有行
        for (int[] row : mat) {
        	// 获取数值和
            int s = binarySearch(row, n);
            // 将索引和数值和加入集合
            rowList.add(new int[]{index++, s+1});
        }
        // 排序
        Collections.sort(rowList, new Comparator<int[]>(){
        	// 自定义排序规则
            public int compare(int[] a, int[] b) {
            	// 优先按照数值和升序
                if (a[1] != b[1]) {
                    return a[1] - b[1];
                }
                else {
                // 其次按照行数升序
                    return a[0] - b[0];
                }
            }
        });
        int[] ans = new int[k];
        // 将排序后前k个索引存入数组
        for (int i = 0; i < k; i++) {
            int[] row = rowList.get(i);
            ans[i] = row[0];
        }
        return ans;
    }

    public int binarySearch(int[] array, int n) {
        int high = n - 1;
        int low = 0;
        // 初始认为没有1，设位置为-1
        int pos = -1;
        // 注意到，low和high重叠时的位置可能也为1，需要进行判断
        while (low <= high) {
            int mid = (low + high) / 2;
            // 当前mid对应位置为0，二分查找左半段
            if (array[mid] == 0)
                high = mid-1;
            else {
            // 当前mid对应位置为1，二分查找右半段，并存下当前位置
                pos = mid;
                low = mid+1;
            }
        }
        return pos;
    }