华为机考真题 -- 智能成绩表

题目描述：

小明来到某学校当老师，需要将学生按考试总分或单科分数进行排名，你能帮帮他吗?

输入描述：

第1行输入两个整数，学生人数n和科目数量m，0<n<100,0<m<10；

第2行输入m个科目名称，彼此之间用空格隔开。科目名称只由英文字母构成，单个长度不超过10个字符。科目的出现顺序和后续输入的学生成绩一一对应。不会出现重复的科目名称；

第3行开始的n行，每行包含一个学生的姓名和该生m个科目的成绩(空格隔开)，学生不会重名。学生姓名只由英文字母构成，长度不超过10个字符。成绩是0~100的整数，依次对应第2行中输入的科目。

第n+2行，输入用作排名的科目名称。若科目不存在，则按总分进行排序。

输出描述：

输出一行，按成绩排序后的学生名字，空格隔开。成绩相同的按照学生姓名字典顺序排序。

示例1：

输入
3 2
yuwen shuxue
fangfang 95 90
xiaohua 88 95
minmin 100 82
shuxue

输出：
xiaohua fangfang minmin

说明：按照shuxue成绩排名，依次是xiaohua fangfang minmin

示例2：

输入
3 2
yuwen shuxue
fangfang 95 90
xiaohua 88 95
minmin 100 82
shuxue

输出：
fangfang minmin xiaohua

说明：排序科目不存在，按总分排序，fangfang和minmin总分相同，按照姓名的字典顺序，fangfang排在前面，所以排名一次是fangfang minmin xiaohua

C++源码：

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>
#include <numeric>
using namespace std;

struct Student {
    string name;
    vector<int> scores;
    int totalScore;

    Student(string n, vector<int> s) : name(n), scores(s) {
        totalScore = accumulate(s.begin(), s.end(), 0);
    }
};

bool compareBySubject(const Student& a, const Student& b, const string& subject) {
    for (size_t i = 0; i < a.scores.size(); ++i) {
        if (a.scores[i] != b.scores[i]) {
            return a.scores[i] < b.scores[i];
        }
    }
    return a.name < b.name; // If scores are the same, sort by name.
}

bool compareByTotal(const Student& a, const Student& b) {
    if (a.totalScore != b.totalScore) {
        return a.totalScore > b.totalScore;
    }
    return a.name < b.name;
}

int main() {
    int n, m;
    cin >> n >> m;

    vector<string> subjects(m);
    for (int i = 0; i < m; ++i) {
        cin >> subjects[i];
    }

    vector<Student> students;
    for (int i = 0; i < n; ++i) {
        string name;
        vector<int> scores(m);
        cin >> name;
        for (int j = 0; j < m; ++j) {
            cin >> scores[j];
        }
        students.emplace_back(name, scores);
    }

    string rankSubject;
    cin >> rankSubject;

    bool sortBySubject = find(subjects.begin(), subjects.end(), rankSubject) != subjects.end();

    if (sortBySubject) {
        sort(students.begin(), students.end(), [&](const Student& a, const Student& b) {
            return compareBySubject(a, b, rankSubject);
            });
    }
    else {
        sort(students.begin(), students.end(), compareByTotal);
    }

    for (const auto& student : students) {
        cout << student.name << " ";
    }
    cout << endl;

    system("pause");
    return 0;
}