【华为OD机试】 选修课

题目描述
现有两门选修课，每门选修课都有一部分学生选修，每个学生都有选修课的成绩，需要你找出同时选修了两门选修课的学生，先按照班级进行划分，班级编号小的先输出，每个班级按照两门选修课成绩和的降序排序，成绩相同时按照学生的学号升序排序。

输入描述
第一行为第一门选修课学生的成绩，

第二行为第二门选修课学生的成绩，

每行数据中学生之间以英文分号分隔，每个学生的学号和成绩以英文逗号分隔，

学生学号的格式为8 88位数字

2位院系编号+入学年份后2位+院系内部1位专业编号+所在班级3位学号

学生成绩的取值范围为[0 , 100 0,1000,100]之间的整数，

两门选修课选修学生数的取值范围为[1 − 2000 1-20001−2000]之间的整数。

输出描述
同时选修了两门选修课的学生的学号，如果没有同时选修两门选修课的学生输出NULL,

否则，先按照班级划分，班级编号小的先输出，每个班级先输出班级编号(学号前五位)，

然后另起一行输出这个班级同时选修两门选修课的学生学号，学号按照要求排序(按照两门选修课成绩和的降序，成绩和相同时按照学号升序学生之间以英文分号分隔。

样例
输入

01202021,75;01201033,95;01202008,80;01203006,90;01203088,100
01202008,70;01203088,85;01202111,80;01202021,75;01201100,88

输出

01202
01202008;01202021
01203
01203088

说明

同时选修了两选修课的学生01202021 、 01202008 、 01203088 01202021、01202008、0120308801202021、01202008、01203088，这三个学生两门选修课的成绩和分别为150 、 150 、 185 150、150、185150、150、185，

01202021 、 01202008 01202021、0120200801202021、01202008届于01202 0120201202班的学生，按照成绩和降序，成绩相同时按学号升序输出的结果为01202008 ; 01202021 01202008;0120202101202008;01202021,

01203088 0120308801203088属于01203 0120301203班的学生，按照成绩和降序，成绩相同时按学号升序输出的结果为0120308 01203080120308，

01202 0120201202的班级编号小于01203 0120301203的班级编号，需要先输出。

输入

01201022,75;01202033,95;01202018,80;01203006,90;01202066,100
01202008,70;01203102,85;01202111,80;01201021,75;01201100,88

输出

NULL

说明

没有同时选修了两门选修课的学生，输出NULL
 

代码：

#include <bits/stdc++.h>
using namespace std;

struct student {
    string num;
    int fs;
};

vector<student> ac;

bool cmp1( student A,  student B) {
    return A.num < B.num;
}

bool cmp2( student A,  student B) {
    string s1 = A.num.substr(0, 5), s2 = B.num.substr(0, 5);
    if (s1 == s2) {
        if (A.fs == B.fs) {
            return A.num < B.num;
        }
        return A.fs > B.fs;
    }
    return s1 < s2;
}

int main() {

    int c = 0;
    string s;

    for (int i = 0; i < 2; i++) {
        cin >> s;
        s += ";";
        vector<string> vs;
        int len = s.size();

        string tmp;
        for (int i = 0; i < len; ++i) {
            if (s[i] == ';') {
                vs.emplace_back(tmp);
                tmp = "";
            } else {
                tmp += s[i];
            }
        }

        for (string v : vs) {
            int p = v.find(",");
            int a;
            istringstream(v.substr(p + 1)) >> a;
            ac.push_back({v.substr(0, p), a});
        }
    }

    sort(ac.begin(), ac.end(), cmp1);

    vector<student> students;
    for (int i = 0; i < ac.size(); ++i) {
        int j = i;
        student student = {ac[i].num, 0};
        int cnt = 0;
        
        while (j < ac.size() && ac[j].num == ac[i].num) {
            student.fs += ac[j].fs;
            cnt += 1;
            j += 1;
        }

        if (cnt == 2) {
            students.push_back(student);
        }

        i = j - 1;
    }


    if (students.empty()) {
        cout << "NULL" << endl;
        return 0;
    }

    sort(students.begin(), students.end(), cmp2);

    // 按照题目要求输出
    for (int i = 0; i < students.size(); ++i) {
        // 先输出每个班级号
        string t = students[i].num.substr(0, 5);
        cout << t << "" << endl;

        int j = i + 1;
        while (j < students.size() && students[j].num.substr(0, 5) == t) {
            j += 1;
        }

        for (int k = i; k < j; ++k) {
            if (k != i) {
                cout << ";";
            }
            cout << students[k].num;
        }
        cout << endl;
        i = j - 1;
    }

    return 0;
}