1. 題目

給你一個(gè)大小為 m * n 的方陣 mat，方陣由若干軍人和平民組成，分別用 0 和 1 表示。

請(qǐng)你返回方陣中戰(zhàn)斗力最弱的 k 行的索引，按從最弱到最強(qiáng)排序。

如果第 i 行的軍人數(shù)量少于第 j 行，或者兩行軍人數(shù)量相同但 i 小于 j，那么我們認(rèn)為第 i 行的戰(zhàn)斗力比第 j 行弱。

軍人 總是 排在一行中的靠前位置，也就是說(shuō) 1 總是出現(xiàn)在 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] 解釋： 每行中的軍人數(shù)目： 行 0 -> 2 行 1 -> 4 行 2 -> 1 行 3 -> 2 行 4 -> 5 從最弱到最強(qiáng)對(duì)這些行排序后得到 [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] 解釋： 每行中的軍人數(shù)目： 行 0 -> 1 行 1 -> 4 行 2 -> 1 行 3 -> 1 從最弱到最強(qiáng)對(duì)這些行排序后得到 [0,2,3,1]提示： m == mat.length n == mat[i].length 2 <= n, m <= 100 1 <= k <= m matrix[i][j] 不是 0 就是 1

來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/the-k-weakest-rows-in-a-matrix
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

2. 優(yōu)先隊(duì)列解題

class Solution { public:vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {int i, j, count;priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>> q;//按pair的first排序，相同則按second，greater是從小到大for(i = 0; i < mat.size(); i++){count = 0;for(j = 0; j < mat[i].size(); j++){if(mat[i][j])count++;}q.push({count, i});//先按軍人數(shù)，再按序號(hào)}vector<int> ans;while(k--){ans.push_back(q.top().second);q.pop();}return ans;} };

總結(jié)

以上是生活随笔為你收集整理的LeetCode 1337. 方阵中战斗力最弱的 K 行（优先队列）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。