Idea

使用C++ STL中的優先隊列
因為創建的優先隊列默認建立的是大頂堆。
所以我們最后存放的答案數組還需要進行一次反轉。

AC Code

class Solution {
public:vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {priority_queue<pair<int,int>> q;int m = mat.size();int n = mat[0].size();for(int i = 0;i < m;i++){int cnt = 0;for(int j = 0;j < n;j++){if(mat[i][j] == 0) break;cnt++;}q.push({cnt,i});}while(q.size() > k) q.pop();vector<int> res;while(q.size()){res.emplace_back(q.top().second);q.pop();}reverse(res.begin(),res.end());return res;}
};