个人任务7.矩阵中战斗力最弱的 K 行

作者: Turbo 时间限制: 1S 章节: 课程设计

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

输入说明：
输入若干行：
第一行输入两个整数m和n，表示矩阵的行数和列数。
之后m行，每行输入n个整数（0或1）表示矩阵的元素。
最后一行输入一个整数k(1 <= k <= m).
提示：
2 <= n, m <= 100
1 <= k <= m
矩阵的元素 不是 0 就是 1

输出说明：
输出一行k个整数，每个整数后跟一个空格。

输入范例：
4 4
1 0 0 0
1 1 0 0
1 0 0 0
1 1 1 0
3

输出范例：
0 2 1

#include<iostream>
#include<vector>
#include<string>
#include<sstream>
#include<queue>
#include<stack>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<stdlib.h>
#include<stdio.h>
#include<fstream>
#include<iomanip>
#include<utility>
using namespace std;
int a[101]={0};//a数组用来记录一行中军人的数量
int b[101][101]={0};

bool cmp(pair<int,int> we,pair<int,int> wk){
    return we.second<wk.second;//根据second的值升序排列
}

int main(){
    int n,m,k;//k表示要输出排序数组的最弱的前几行
    cin>>n>>m;
    for(int i=0;i<n;++i){
        for(int j=0;j<m;++j){
            cin>>b[i][j];//输入n行m列的二维数组
            if(b[i][j]==1)
                a[i]++;//在输入的过程中就计数
        }
    }
    cin>>k;
    pair<int,int> dui[n];
    for(int i=0;i<n;++i)
        dui[i]=make_pair(i,a[i]);//将对应序号和军人个数形成对组
    stable_sort(dui,dui+n,cmp);
    for(int i=0;i<k;++i)
        cout<<dui[i].first<<" ";
    return 0;
}