Leetcode--Java--1337. 矩阵中战斗力最弱的 K 行

题目描述

给你一个大小为 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]

思路

1. 暴力做法。用二维数组来存储，其中每个元素为一个二元组，只存两个元素：每行军人数目和索引。
2. 遍历原二维数组，记录每行的军人数目以及对应索引值，作为整体存入一个新的二维数组里面。
3. 排序时指定优先级

代码

class Solution {
    public int[] kWeakestRows(int[][] mat, int k) {
        int res[] = new int[k];
        int m = mat.length, n = mat[0].length;
        //构造一个二维数组 每个数组元素有两个值：每行军人数目和索引值
        int all[][] = new int[m][2];
        for (int i = 0; i < m; i ++ ) {
            int cnt = 0; //军人数目
            for (int j = 0; j < n; j ++ ) {
                cnt += mat[i][j];
            }
            //构造二元组
            all[i] = new int[]{cnt, i};
        }
        //排序
        Arrays.sort(all, (a, b) -> {
            //优先按军人数目，再按索引
           if (a[0] != b[0]) {
               return a[0] - b[0];
           }
           else return a[1] - b[1];
        });
        //最小的k行
        for (int i = 0; i < k; i ++ ) {
            res[i] = all[i][1];
        }
        return res;
    }
}