本文主要是介绍1337. 方阵中战斗力最弱的 K 行,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
模拟
矩阵从小到大排序,当1的数字相同时,索引小的行排在前面,所以在矩阵的最后一列增加一个当前所在行的所以,直接用vector内置的比较函数进行比较。最后统计结果时,直接返回最后一行的索引。
class Solution {
public:vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {int m = mat.size();if(!m) return {};int n = mat[0].size();for(int i = 0; i < m; i++) mat[i].push_back(i);sort(mat.begin(), mat.end());vector<int> res;for(int i = 0; i < k; i++){res.push_back(mat[i][n]);}return res;}
};
这篇关于1337. 方阵中战斗力最弱的 K 行的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!