1095 解码PAT准考证

/**
 * 1.解题思路：给出一组学生的准考证号和成绩，准考证号包含了等级(乙甲顶)，考场号，日期，和个人编号信息，并有三种查询方式
 *            查询一：给出考试等级，找出该等级的考生，按照成绩降序，准考证升序排序
 *            查询二：给出考场号，统计该考场的考生数量和总得分
 *            查询三：给出考试日期，查询改日期下所有考场的考试人数，按照人数降序，考场号升序排序
 *            先把所有考生的准考证和分数记录下来:
 *            1.按照等级查询，枚举选取匹配的学生，然后排序即可
 *            2.按照考场查询，枚举选取匹配的学生，然后计数、求和
 *            3.按日期查询每个考场人数，用unordered_map存储，最后排序汇总
 *            
 * 2.注意：1.第三个用map存储会超时，用unordered_map就不会超时
 *        2.排序传参建议用引用传参，这样更快，虽然有时候不用引用传参也能通过，但还是尽量用
 * 
 * 3.参考博客：https://blog.csdn.net/liuchuo/article/details/84972869
 **/
#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
using namespace std;
//准考证号 成绩
struct node {
    string t;
    int value;
};
bool cmp(const node &a, const node &b) {
    return a.value != b.value ? a.value > b.value : a.t < b.t;
}
int main() {
    int n, k, num;
    string s;
    cin >> n >> k;
    vector<node> v(n);
    for (int i = 0; i < n; i++)
        cin >> v[i].t >> v[i].value;
    // k行统计要求
    for (int i = 1; i <= k; i++) {
        cin >> num >> s;
        //c_str()函数返回一个指向正规C字符串的指针, 内容与本string串相同.
        printf("Case %d: %d %s\n", i, num, s.c_str());
        vector<node> ans;
        int cnt = 0, sum = 0;
        //查询一:按分数非升序输出某个指定级别的考生的成绩
        if (num == 1) {
            for (int j = 0; j < n; j++)
                //找出与s[0]相同的等级
                if (v[j].t[0] == s[0]) ans.push_back(v[j]);
        //查询二：将某指定考场的考生人数和总分统计输出
        } else if (num == 2) {
            for (int j = 0; j < n; j++) {
                if (v[j].t.substr(1, 3) == s) {
                    cnt++;
                    sum += v[j].value;
                }
            }
            if (cnt != 0) printf("%d %d\n", cnt, sum);
        //查询三:将某指定日期的考生人数分考场统计输出
        } else if (num == 3) {
            //用unordered_map就不会超时
            unordered_map<string, int> m;
            for (int j = 0; j < n; j++)
                if (v[j].t.substr(4, 6) == s) m[v[j].t.substr(1, 3)]++;
            for (auto it : m) ans.push_back({it.first, it.second});
        }
        sort(ans.begin(), ans.end(),cmp);
        //输出
        for (int j = 0; j < ans.size(); j++) printf("%s %d\n", ans[j].t.c_str(), ans[j].value);
        //查询结果为空，则输出 NA
        if (((num == 1 || num == 3) && ans.size() == 0) || (num == 2 && cnt == 0)) printf("NA\n");
    }
    return 0;
}