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

• 题目描述
• 解题思路
• 执行结果

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

题目描述

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] 示例 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]

提示：

m == mat.length n == mat[i].length 2 <= n, m <= 100 1 <= k <= m matrix[i][j] 不是 0 就是 1

解题思路

法1

使用一个优先队列（PriorityQueue）来存储最弱的k行的信息。优先队列按照自定义的比较器来进行排序，比较器中首先比较军人数量，如果数量不同则按数量降序排序，如果数量相同则按行索引降序排序。

然后遍历矩阵的每一行，统计军人数量，并将数量和行索引加入优先队列中。同时，保持队列的大小为k，如果队列大小超过k，则移除队列中的最强行。

最后，将优先队列中的结果按顺序取出，存入结果数组中，并返回结果数组。

• 时间复杂度(O(N))
• 空间复杂度(O(N))

执行结果

法1

class Solution {
    public int[] kWeakestRows(int[][] mat, int k) {
        int m = mat.length;
        int n = mat[0].length;

        // 创建一个优先队列，用于按战斗力从最弱到最强排序
        PriorityQueue<int[]> pq = new PriorityQueue<>(k, (a, b) -> {
            if (a[0] != b[0]) {
                return b[0] - a[0]; // 按军人数量降序排序
            } else {
                return b[1] - a[1]; // 如果军人数量相同，按行索引降序排序
            }
        });

        // 遍历矩阵，统计每行的军人数量并加入优先队列
        for (int i = 0; i < m; i++) {
            int count = countSoldiers(mat[i]);
            pq.offer(new int[]{count, i});
            if (pq.size() > k) {
                pq.poll(); // 保持队列的大小为k
            }
        }

        // 将优先队列中的结果按顺序取出
        int[] result = new int[k];
        while (!pq.isEmpty()) {
            result[--k] = pq.poll()[1];
        }

        return result;
    }

    // 辅助函数，统计数组中军人的数量
    private int countSoldiers(int[] row) {
        int count = 0;
        for (int num : row) {
            if (num == 1) {
                count++;
            } else {
                break; // 遇到第一个0，后面的元素必然都是0
            }
        }
        return count;
    }

}

执行结果： 通过 显示详情 查看示例代码 添加备注

执行用时： 1 ms , 在所有 Java 提交中击败了 96.42% 的用户 内存消耗： 43.4 MB , 在所有 Java 提交中击败了 45.34% 的用户 通过测试用例： 52 / 52

本文由 mdnice 多平台发布