B3827 [NICA #2] 高考组题

题目描述

高考是能决定每个考生命运的考试，因此作为组卷人的 Aya 将高考组卷视为重中之重。现在 Aya 有 $n$ 个备选的高考题。

对于每一道可能选入试卷的高考题，其有 $k$ 个指标来评判它各个方面的维度，例如难度、新颖度、与现实生活的结合性等。每个指标均为一个正整数。

Aya 认为，一个题被放入高考卷的必要性，是这些指标的平均值。他认为，必要性前二高的试题必须要被放进高考试卷。若有多个必要性前二大的试题则选择较早出现的那个。请告诉 Aya 哪个题目是必须要放进高考试卷的。

输入格式

第一行输入两个正整数 $n,k$，含义如题所述；

第二行开始，往下 $n$ 行，每行输入 $k$ 个正整数，以空格隔开，第 $i$ 行表示第 $i-1$ 道题的各个指标是多少。

输出格式

输出两行，表示必要性前两大的试题是哪两个题。优先输出必要性最高的；若有多个题目必要性相同，则优先输出较早出现的。

样例 #1

样例输入 #1

3 2
10 20
15 25
20 30

样例输出 #1

3
2

样例 #2

样例输入 #2

3 2
10 20
5 25
1 2

样例输出 #2

1
2

提示

数据保证，$2\le n\le 100$，$1\le k\le 10$，$1\le$ 一道试题的各个指标 $\le 100$。

思路：把每个题的指标加在一起，再按照这个指标排序好，再取前两个就行

#include <algorithm>
#include<iostream>
using namespace std;
struct node {
	int index, sum;
} arr[105];
int n, k;
bool cmp(node a, node b) {
	return a.sum > b.sum || (a.sum == b.sum && a.index < b.index);
}
int main(){
	cin >> n >> k;
	for(int i = 1; i <= n; i++) {
		int sum = 0;
		arr[i].index = i;
		for(int j = 1; j <= k; j++) {
			int a;
			cin >> a;
			sum += a;
		}
		arr[i].sum = sum;
	}
	sort(arr+1, arr+n+1, cmp);
	cout << arr[1].index << "\n" << arr[2].index;
	return 0;
}