【PAT乙级】解码PAT准考证（ C 类型输入输出，unordered_map 嵌套 ，字符串处理，结构体排序）

狠狠优化后

PAT 准考证号由 4 部分组成：

• 第 1 位是级别，即 T 代表顶级；A 代表甲级；B 代表乙级；
• 第 2~4 位是考场编号，范围从 101 到 999；
• 第 5~10 位是考试日期，格式为年、月、日顺次各占 2 位；
• 最后 11~13 位是考生编号，范围从 000 到 999。

现给定一系列考生的准考证号和他们的成绩，请你按照要求输出各种统计信息。

输入格式：

输入首先在一行中给出两个正整数 N（≤10​4​​）和 M（≤100），分别为考生人数和统计要求的个数。

接下来 N 行，每行给出一个考生的准考证号和其分数（在区间 [0,100] 内的整数），其间以空格分隔。

考生信息之后，再给出 M 行，每行给出一个统计要求，格式为：类型 指令，其中

• 类型 为 1 表示要求按分数非升序输出某个指定级别的考生的成绩，对应的 指令 则给出代表指定级别的字母；
• 类型 为 2 表示要求将某指定考场的考生人数和总分统计输出，对应的 指令 则给出指定考场的编号；
• 类型 为 3 表示要求将某指定日期的考生人数分考场统计输出，对应的 指令 则给出指定日期，格式与准考证上日期相同。

输出格式：

对每项统计要求，首先在一行中输出 Case #: 要求，其中 # 是该项要求的编号，从 1 开始；要求 即复制输入给出的要求。随后输出相应的统计结果：

• 类型 为 1 的指令，输出格式与输入的考生信息格式相同，即 准考证号 成绩。对于分数并列的考生，按其准考证号的字典序递增输出（题目保证无重复准考证号）；
• 类型 为 2 的指令，按 人数 总分 的格式输出；
• 类型 为 3 的指令，输出按人数非递增顺序，格式为 考场编号 总人数。若人数并列则按考场编号递增顺序输出。

如果查询结果为空，则输出 NA。

输入样例：

8 4
B123180908127 99
B102180908003 86
A112180318002 98
T107150310127 62
A107180908108 100
T123180908010 78
B112160918035 88
A107180908021 98
1 A
2 107
3 180908
2 999

输出样例：

Case 1: 1 A
A107180908108 100
A107180908021 98
A112180318002 98
Case 2: 2 107
3 260
Case 3: 3 180908
107 2
123 2
102 1
Case 4: 2 999
NA

思路

我本来想昨天晚上做，但是看时间太晚了就只读了读题。认为这跟普通的结构体排序一样，谁知道今天弄了半天。有思路，但是死活不对，后来知道哪里错了后，第三个测试点总是超时。把 cin 同步取消掉也不行。于是就开始了优化指路。

• 第一种情况的优化，开了4个 vector ，一个存储所有结构体。其他三个分别存储 'A' 'B' 'T' ，对这三种结构体排序。
• 第二种情况，开了一个二维数组存储，以考场号为下表，存储人数和总分。在输入的过程中，对数组进行赋值，用 set 确认数组中有这个考场号。
• 第三种情况，也是最复杂的一种。map 嵌套，声明一个二维 map ，key 决定对应日期的考场+人数的 map
• 输入输出优化：所有的输出都用的 printf 。如果用 cout 把同步关了也会超时。
• 因为 map 与 set 都有排序的功能，而我们这道题用不到。因此用的都是未排序版的 map 与 set
• 函数都用的引用传递

#include
#include 
#include 
#include
#include
#include 
#include 
using namespace std;
int type2[1005][2];
unordered_set ss;
unordered_map >ultimate;
struct student
{
	string ID;
	int score;
	string leixing;
	string date;
	string kaochang;
	void setting()
	{
		leixing = ID[0];
		kaochang = ID.substr(1, 3);
		date = ID.substr(4, 6);
		ss.insert(kaochang);
		ultimate[date][kaochang]++;
		type2[stoi(kaochang)][0]++;
		type2[stoi(kaochang)][1] += score;
	}
};
bool cmp_1(student &a, student & b)
{
	if (a.score != b.score)
		return a.score > b.score;
	else
		return a.ID < b.ID;
}
bool cmp(pair &a, pair &b)
{
	if (a.second != b.second)
		return a.second > b.second;
	else
		return a.first < b.first;
}
vector vec;
vector Tvec;
vector Avec;
vector Bvec;
void out(const vector & a,bool & have)
{
	
	int size = a.size();
	for (int i = 0; i < size; i++)
	{
		have = true;
		printf("%s %d\n", a[i].ID.c_str(), a[i].score);
	}		
}
int main() {
	int N, M;
	scanf("%d %d", &N, &M);
	for (int i = 0; i < N; i++)
	{
		student temp;
		cin >> temp.ID >> temp.score;
		temp.setting();
		vec.push_back(temp);
		if (temp.ID[0] == 'T') Tvec.push_back(temp);
		else if (temp.ID[0] == 'A') Avec.push_back(temp);
		else  Bvec.push_back(temp);
	}
	sort(Tvec.begin(), Tvec.end(), cmp_1);
	sort(Avec.begin(), Avec.end(), cmp_1);
	sort(Bvec.begin(), Bvec.end(), cmp_1);
	for (int q = 1; q <= M; q++)
	{
		int type; string instruction;
		cin >> type >> instruction;
		int size = vec.size();
		printf("Case %d: %d %s\n", q, type, instruction.c_str());
		bool have = false;
		if (type == 1)
		{			
			if (instruction == "A") out(Avec,have);
			else if (instruction == "T") out(Tvec,have);
			else if (instruction == "B") out(Bvec,have);
		}
		else if (type == 2)
		{
			//此时的 instruction 为考场
			if (ss.count(instruction)) //检查考场是否存在
			{
				have = true;
				int namekaoschang = stoi(instruction);
				printf("%d %d\n", type2[namekaoschang][0], type2[namekaoschang][1]);
			}
		}
		else if(type==3)
		{
			vector > temp(ultimate[instruction].begin(), ultimate[instruction].end());
			sort(temp.begin(), temp.end(),cmp);
			for (auto it = temp.begin(); it !=temp.end(); it++)
			{
				have = true;
				printf("%s %d\n", it->first.c_str(), it->second);
			}
		}
		if (!have)
			cout << "NA" << endl;
	}
}

反思

谨慎使用全局变量！