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;
    }
};