机试题——拔河比赛队员选择

题目描述

某团队需要组织一支队伍参加拔河比赛，团队共有队员 ( n ) 人，比赛队员人数要求为 ( m ) 人，满足 ( n > m )。团队成员按编号 1 到 ( n ) 的顺序参加了 ( k ) 轮力量测试，每轮的测试成绩用正整数表示。

根据 ( n ) 个队员的力量测试成绩选择比赛队员 ( m ) 人。选择规则如下：

1. 先选择 ( k ) 轮测试中最好成绩最大的队员。
2. 若有多人的最好成绩相等，则优先选择其中第二好成绩最大的队员，依次类推。
3. 若还有相等的情况，则优先选择编号较小的队员。
   每个人只能被选择一次。

输入描述

第一行包含三个正整数 ( n )、( m ) 和 ( k )，分别表示团队队员总数、比赛队员要求的人数和力量测试轮数。

接下来的 ( n ) 行，每行包含 ( k ) 个正整数，表示第 ( i ) 个队员在 ( k ) 轮测试中的成绩。

满足：

• ( 0 < m < n < 10^3 )
• ( 0 < k < 10^3 )
• 每轮测试成绩 ( 0 < 成绩 < 10^5 )

输出描述

输出按上述选择顺序选出的比赛队员编号的列表，用空格分隔。

用例输入

样例 1

输入：

4 3 3
10 12 14
11 12 13
12 15 10
12 11 13

输出：

3 1 2

说明：
团队有队员 4 人，比赛队员需 3 人，经过 3 轮测试。队员 1~4 的测试最大值分别为 14、13、15、13，队员 3 和 1 的测试最大值为 15 和 14 较大，因此先选择队员 3 和 1。队员 2 和 4 的成绩从大到小排序后都相等，因此选序号较小的队员 2，输出为 3 1 2。

解题思路

问题分析

1. 目标：根据队员的测试成绩，按照特定规则选择 ( m ) 名队员。
2. 关键点：
   • 需要比较每个队员的多轮测试成绩。
   • 成绩的优先级从高到低依次比较。
   • 如果成绩完全相同，优先选择编号较小的队员。

算法设计

1. 数据结构：

   • 使用结构体 node 存储每个队员的信息，包括编号和排序后的成绩列表。

2. 排序规则：

   • 自定义比较函数 cmp，用于比较两个队员的成绩。
   • 遍历每个队员的成绩列表，从高到低依次比较。
   • 如果成绩完全相同，比较队员编号。

3. 实现步骤：

   • 输入 ( n )、( m ) 和 ( k )。
   • 读取每个队员的 ( k ) 轮测试成绩，并存储到结构体中。
   • 对每个队员的成绩列表进行降序排序。
   • 使用自定义比较函数对所有队员进行排序。
   • 输出前 ( m ) 名队员的编号。

代码实现

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

struct node {
    int id; // 队员编号
    vector<int> scores; // 排序后的成绩列表
};

// 自定义比较函数
bool cmp(const node& a, const node& b) {
    // 比较每个成绩
    for (int i = 0; i < a.scores.size(); ++i) {
        if (a.scores[i] != b.scores[i]) {
            return a.scores[i] > b.scores[i]; // 成绩高者优先
        }
    }
    // 成绩完全相同，比较编号
    return a.id < b.id;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m, k;
    cin >> n >> m >> k; // 输入 n, m, k
    vector<node> datas(n);

    for (int i = 0; i < n; ++i) {
        datas[i].id = i + 1; // 编号从 1 开始
        datas[i].scores.resize(k);
        for (int j = 0; j < k; ++j) {
            cin >> datas[i].scores[j]; // 输入每轮成绩
        }
        // 将成绩排序降序
        sort(datas[i].scores.begin(), datas[i].scores.end(), greater<int>());
    }

    // 对所有队员排序
    sort(datas.begin(), datas.end(), cmp);

    // 输出前 m 个队员的编号
    for (int i = 0; i < m; ++i) {
        if (i > 0) cout << ' ';
        cout << datas[i].id;
    }

    return 0;
}