ARST打卡第117周[117/521]

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

class Solution {
public:
    static bool comp(const pair<int, int> a, const pair<int, int> b) {
        if (a.first != b.first) {
            return a.first < b.first;
        }
        return a.second < b.second;
    }

    // 2021年08月01日12:41 应该就是把 count, lien_id绑定,然后排序
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        vector<pair<int, int> > tmp;
        for (int i = 0; i < mat.size(); i++) {
            int lien_count = 0;
            for (auto x : mat[i]) {
                lien_count += x;
            }
            tmp.push_back({lien_count, i});
        }

        sort(tmp.begin(), tmp.end(), comp);

        vector<int> ans;
        for (int i = 0; i < k; i++) {
            ans.push_back(tmp[i].second);
        }

        return ans;
    }
};