算法刷题笔记

news2024/11/24 20:08:26

1.力扣-1337.矩阵中战斗力最弱的K行

给你一个大小为 m * n 的矩阵 mat，矩阵由若干军人和平民组成，分别用 1 和 0 表示。
请你返回矩阵中战斗力最弱的 k 行的索引，按从最弱到最强排序。
如果第 i 行的军人数量少于第 j 行，或者两行军人数量相同但 i 小于 j，那么我们认为第 i 行的战斗力比第 j 行弱。
军人总是排在一行中的靠前位置，也就是说 1 总是出现在 0 之前。
示例 1：
输入：mat =
[[1,1,0,0,0],
[1,1,1,1,0],
[1,0,0,0,0],
[1,1,0,0,0],
[1,1,1,1,1]],
k = 3
输出：[2,0,3]
解释：
每行中的军人数目：
行 0 -> 2
行 1 -> 4
行 2 -> 1
行 3 -> 2
行 4 -> 5
从最弱到最强对这些行排序后得到 [2,0,3,1,4]
示例 2：
输入：mat =
[[1,0,0,0],
[1,1,1,1],
[1,0,0,0],
[1,0,0,0]],
k = 2
输出：[0,2]
解释：
每行中的军人数目：
行 0 -> 1
行 1 -> 4
行 2 -> 1
行 3 -> 1
从最弱到最强对这些行排序后得到 [0,2,3,1]
提示：
m == mat.length
n == mat[i].length
2 <= n, m <= 100
1 <= k <= m
matrix[i][j] 不是 0 就是 1

答案

 vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        int m = mat.size();
        int n = mat[0].size();
        vector<pair<int,int>> midv;

        for(int i=0;i<m;i++){
            int l=0,r=n-1,pos=-1;
            while(l<=r){
                int mid = l+(r-l)/2;
                if(mat[i][mid]==0){
                    r = mid -1;
                }else{
                    pos = mid;
                    l = mid + 1;
                }
            }
            midv.emplace_back(pos+1,i);
        }
        sort(midv.begin(),midv.end());
        vector<int> v;
        for(int i=0;i<k;i++){
            v.push_back(midv[i].second);
        }
        return v;
    }

知识点1

二维数组默认sort及自定义sort函数

//sort对二维数组排序规则- 按每一行规则排序，默认从小到大，先比第一列，也就是每行的第一个，小的则会移动到最前边，后边依次。 
示例：
bool cmp(vector<int>x, vector<int>y){
	return x[0]>y[0]||x[0]==y[0];
}
void show1(vector<int>v){
    cout<<v[0]<<" "<<v[1]<<" "<<v[2]<<endl;
}
int main() {
	vector<vector<int>>nums={{2,1,3},{10,4,2},{2,0,6}};
	//若map中加入 cmp 则会按每一行指定的列形式排序。此时是x[0]，那么就是按照每一行的第0列进行比较排序,x[0]>y[0]就是递减排序，就不会比较其他列.
	sort(nums.begin(),nums.end());
	for_each(nums.begin(),nums.end(),show1);
	return 0;
}

打印结果：此时是先根据每一行的将第一列进行比较，对2 0 6 和 2 1 3，第一列的2一样，因此比较第二列的0 和 1，0小于1，所以2 0 6在2 1 3前边，同理，2=2<10，因此10 是在最后一行。

加入cmp函数：按照每一行的第一列进行排序
在这里插入图片描述

知识点2

# 键值对形式不仅有map,vector也可以使用，如：
	vector<pair<int,int>> midv;
# 其添加数据的形式同vector：
	midv.emplace_back(pos+1,i);
# 取数据同vector:
	for(int i=0;i<midv.size();i++){
		cout<<midv[i].first<<"-"<<midv[i].second<<endl;
	}
# 其使用sort的规则同上述二维数组
	sort(midv.begin(),midv.end());