PAT乙级 1041 考试座位号

每个 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

第一种代码：

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

// 初始化数组 
long long testid[1000][3]={0};
int checkid[1000]={0};
int main()
{
    int A,B;
    cin>>A;
    for(int i=0;i<A;i++)
    {
        for(int j=0;j<3;j++)
        {
            cin>>testid[i][j];
        }
        //cin>>endl;
    }
    cin>>B;
    for(int i=0;i<B;i++)
    {
        cin>>checkid[i];
    }
    for(int k=0;k<B;k++) // 注意次数的循环顺序 
    {
        for(int i=0;i<A;i++)
        {
            if(checkid[k]==testid[i][1])
            {
                cout<<testid[i][0]<<" "<<testid[i][2];
                break;
            }
        }
        cout<<endl;
    }
    return 0;
}

第二种代码：

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

struct Student{
	long long id;
	int seat;
};
int main()
{
	int N,M;
	cin>>N;
	map<int,Student>test;
	for(int i=0;i<N;i++)
	{
		long long id;
		int seat_c,seat;
		cin>>id>>seat_c>>seat;
		test[seat_c]={id,seat};
	}
	cin>>M;
	vector<int>check(M); // int check[M];
	for(int i=0;i<M;i++)
	{
		cin>>check[i];
	}
	for(int i=0;i<M;i++)
	{
		int seat_c=check[i];
		cout<<test[seat_c].id<<" "<<test[seat_c].seat<<endl;
	}
	return 0;
 }