作者: 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;
}