LeetCode:1337.方阵中战斗力最弱的k行

题目：

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

请你返回方阵中战斗力最弱的 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]

源码：

class Solution {
    public int[] kWeakestRows(int[][] mat, int k) {
        int len = mat.length; // mat一维数组长度
        int size = mat[0].length; // mat 二维数组的长度
        boolean[] tag = new boolean[len]; // 用来旁段该一维数组是否统计过战斗力
        int[] res = new int[k]; // 保存最终的结果
        int x = 0; // 用来和 k 比较，判断是否结束循环
        for (int i = 0; i < size; i++) {
            for (int j = 0; j < len; j++) {
                // 注意：mat[j][i] 表示在同一个下标下，看哪个一维数组该位置为 0
                // 满足第一个条件：x < k 还没统计完
                // 第二个条件：满足该位置为平民也就是 0
                // 第三个条件：该数组并没有结束战斗力的统计，也就是没有碰到过 0 
                if (x < k && mat[j][i] == 0 && !tag[j]) {
                    res[x] = j;
                    tag[j] = true;
                    x++;
                }
            }
        }
        // 可能出现整个一维数组都是 1 的情况，此时要重新遍历
        int y = 0;
        while (x < k) {
            if (!tag[y]) {
                res[x] = y;
                x++;
                tag[y] = true;
            }
            y++;
        }
        return res;
    }
}