1095 解码PAT准考证 (25分)【解决超时问题技巧】

1095 解码PAT准考证 (25分)

一、题干

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

二、代码

#include <iostream>
#include <algorithm>
#include <vector>
#include<unordered_map>
#include <iomanip>        
#include <string>
using namespace std;

struct stu {
	string hao;
	int fen;
};
struct stu2 {
	int ren, sumfen;
};
struct stu3 {
	int kao, ren;
};

int N, M;
vector<stu> v1, v2, v3;
stu2 st[999] = { stu2{ 0,0 } };
unordered_map<int, vector<stu3>> m;

bool cmp1(const stu &a,const stu &b) {
	if (a.fen != b.fen) return a.fen > b.fen;
	else return a.hao < b.hao;
}

bool cmp3(const stu3 &a,const stu3 &b) {
	if (a.ren != b.ren) return a.ren > b.ren;
	else return a.kao < b.kao;
}
int one(char c) {
	if (c == 'B') {
		for (int i = 0; i < v1.size(); i++) {
			printf("%s %d\n", v1[i].hao.c_str(), v1[i].fen);
		}
		if (v1.size() == 0) printf("NA\n");
		return 0;
	}
	if (c == 'A') {
		for (int i = 0; i < v2.size(); i++) {
			printf("%s %d\n", v2[i].hao.c_str(), v2[i].fen);
		}
		if (v2.size() == 0) printf("NA\n");
		return 0;
	}
	if (c == 'T') {
		for (int i = 0; i < v3.size(); i++) {
			printf("%s %d\n", v3[i].hao.c_str(), v3[i].fen);
		}
		if (v3.size() == 0) printf("NA\n");
		return 0;
	}
}
void two(int hh) {
	if(st[hh].ren!=0) printf("%d %d\n", st[hh].ren,st[hh].sumfen);
	else printf("NA\n");
}
void three(int gg) {
	if (m[gg].size() == 0) {
		printf("NA\n");
		return;
	}
	sort(m[gg].begin(), m[gg].end(), cmp3);
	for (int i = 0; i < m[gg].size(); i++) {
		printf("%d %d\n",  m[gg][i].kao, m[gg][i].ren);
	}
}
int main() {
	scanf("%d %d",&N,&M);
	for (int i = 0; i < N; i++) {
		string s;
		int temp;
		cin >> s >> temp;
		if (s[0] == 'B') v1.push_back(stu{ s,temp });
		if (s[0] == 'A') v2.push_back(stu{ s,temp });
		if (s[0] == 'T') v3.push_back(stu{ s,temp });
		string s2 = s.substr(1, 3);
		int kao = stoi(s2);
		st[kao].ren++;
		st[kao].sumfen += temp;
		string s3 = s.substr(4, 6);
		int data = stoi(s3);
		int flag = 0;
		for (int j = 0; j < m[data].size(); j++) {
			if (m[data][j].kao == kao) {
				m[data][j].ren++;
				flag = 1;
				break;
			}
		}
		if (flag == 0) m[data].push_back(stu3{ kao,1 });
	}
	sort(v1.begin(), v1.end(), cmp1);
	sort(v2.begin(), v2.end(), cmp1);
	sort(v3.begin(), v3.end(), cmp1);
	for (int i = 1; i <= M; i++) {
		int cc;
		cin >> cc;
		if (cc == 1) {
			char c;
			cin >> c;
			cout << "Case " << i << ": " << cc << " " << c << endl;
			one(c);
		}
		if (cc == 2) {
			int hh;
			cin >> hh;
			cout << "Case " << i << ": " << cc << " " << hh << endl;
			two(hh);
		}
		if (cc == 3) {
			int gg;
			cin >> gg;
			cout << "Case " << i << ": " << cc << " " << setfill('0')<<setw(6)<<gg << endl;
			three(gg);
		}
	}
	return 0;
}

三、分析