8-1每日一题：矩阵中战斗力最弱的k行

最新推荐文章于 2025-07-29 16:34:16 发布

原创最新推荐文章于 2025-07-29 16:34:16 发布 · 104 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

每日一题专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种结合二分查找与优先队列的数据结构算法，用于解决矩阵中寻找战斗力最弱的k行的问题。该算法首先通过二分查找确定每行1的数量，再利用优先队列找出具有最少1的前k行。

矩阵中战斗力最弱的k行

二分+优先队列

每行二分查找最后1的位置，优先队列取出前k个出现1次数最小的行号

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        int n = mat.size(), m = mat[0].size();
        vector<pair<int,int>> p;
        for (int i = 0; i < n; i ++ )
        {
            int l = 0, r = m - 1, pos = -1;
            while (l <= r){
                int mid = l + r >> 1;
                if (mat[i][mid] == 0) r = mid - 1;
                else pos = mid, l = mid + 1;
            }
            p.emplace_back(r + 1, i);
        }

        priority_queue q(greater<pair<int, int>>(), move(p));
        vector<int>res;
        for (int i = 0; i < k; i ++ )
        {
            res.emplace_back(q.top().second);
            q.pop();
        }
        return res;
    }
};