1337_TheKWeakestRowsinaMatrix - a920604a/leetcode GitHub Wiki

class Solution {
public:
    int BinarySearch(vector<int> & nums){
        int l = 0, r = nums.size()-1;
        while(l<=r){
            int mid = l + (r-l)/2;
            if(nums[mid] == 0)  r = mid-1;
            else l = mid+1;
        }
        if(l<0) return 0;
        return l+1;
    }
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        // binary search
        int n = mat.size();
        vector<pair<int,int> > ret;
        for(int i=0;i<n;++i){
            int count = BinarySearch(mat[i]);
            ret.push_back({count, i});
        }
        sort(ret.begin(), ret.end(), [](const auto &a, const auto &b){
            if(a.first==b.first) return a.second<b.second;
            else return a.first<b.first;
        });
        vector<int> ans;
        for(int i=0;i<k;++i) ans.push_back(ret[i].second);
        return ans;
    }
};