【PATB1041】考试座位号（题解＋拓展）

原题

题目

每个 PAT 考生在参加考试时都会被分配两个座位号，一个是试机座位，一个是考试座位。正常情况下，考生在入场时先得到试机座位号码，入座进入试机状态后，系统会显示该考生的考试座位号码，考试时考生需要换到考试座位就座。但有些考生迟到了，试机已经结束，他们只能拿着领到的试机座位号码求助于你，从后台查出他们的考试座位号码。

输入格式

输入第一行给出一个正整数 N（≤1000），随后 N 行，每行给出一个考生的信息：准考证号 试机座位号 考试座位号。其中准考证号由 16 位数字组成，座位从 1 到 N 编号。输入保证每个人的准考证号都不同，并且任何时候都不会把两个人分配到同一个座位上。
考生信息之后，给出一个正整数 M（≤N），随后一行中给出 M 个待查询的试机座位号码，以空格分隔。

输出格式

对应每个需要查询的试机座位号码，在一行中输出对应考生的准考证号和考试座位号码，中间用 1 个空格分隔。

输入样例

4
3310120150912233 2 4
3310120150912119 4 1
3310120150912126 1 3
3310120150912002 3 2
2
3 4

输出样例

3310120150912002 2
3310120150912119 1

代码及思路

第一次做时用了遍历的方法，虽然没有超时但是显得不是特别聪明……偶然看到柳神的代码于是醍醐灌顶了~
代码及思路参考柳神blog

#include <iostream>

using namespace std;

int main()
{
    string stu[1010][2], s1, s2;
    int n, test;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> s1 >> test >> s2;
        stu[test][0] = s1;
        stu[test][1] = s2;
    }
    int m;
    cin >> m;
    for (int i = 0; i < m; i++)
    {
        int t;
        cin >> t;
        cout << stu[t][0] << " " << stu[t][1] << endl;
    }
    return 0;
}

拓展（通过准考证号找另两个号）

看了柳神的写法觉得非常妙，于是我思考能否用类似映射的方式通过准考证号找到试机号和座位号。
首先字符串到其他的映射需要用到map，通常采用map<string, int> a;这样的写法，但本题一个准考证号对应着两项内容，即这是字符串到数组的映射，于是我猜想是否能用vector<int>来代表数组，试了一下发现成功了，下面记录一下我的代码：

#include <iostream>
#include <map>
#include <vector>

using namespace std;

int main()
{
	int n;
	cin >> n;
	map<string, vector<int>> map; 
	for (int i = 0; i < n; i++)
	{
		string s;
		int test, real;
		cin >> s >> test >> real;
		map[s].push_back(test);
		map[s].push_back(real);
	}
	int m;
	cin >> m;
	for (int i = 0; i < m; i++)
	{
		string str;
		cin >> str;
		cout << map[str][0] << ' ' << map[str][1] << endl;
	}
}

这样就能通过准考证号查找你的试机号和考试号啦~