LeetCode 1337矩阵中战斗力最弱的K行[自定义排序] HERODING的LeetCode之路

Posted 2021-08-30 HERODING23

解题思路：
一个标准的排序问题，根据下标进行排序，创新点在于排序的内容是与其他数组相关，所以解决方案是我们先获取所有行数的下标，对于下标的排序，通过自定义排序函数进行，代码如下：

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        int len = mat.size();
        vector<int> res;
        for(int i = 0; i < len; i ++) {
            res.push_back(i);
        }
        // 自定义排序
        sort(res.begin(), res.end(), [&](const int& a, const int& b) {
            return mat[a] == mat[b] ? a < b : mat[a] < mat[b];
        });
        vector<int> ans(res.begin(), res.begin() + k);
        return ans;
    }
};