PTA基础题——考试座位号

题设

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

解答

C语言：

#include<stdio.h>
struct In
{
	long long n;
	int s1;
	int s2;
};
int main()
{
	int N;
	scanf("%d",&N);
	In s[1000];
	for(int i=0;i<N;i++)
	{
		scanf("%lld %d %d",&s[i].n,&s[i].s1,&s[i].s2);
	}
    int M;
    scanf("%d",&M);
    int p[1000];
    for(int j=0;j<M;j++)
    {
    	scanf("%d",&p[j]);
    	for(int x=0;x<N;x++)
    	{
    		if(p[j]==s[x].s1)
    		    printf("%lld %d\n",s[x].n,s[x].s2);
		}
	}
	return 0;
}

程序冗长，改进空间较大，在DevC上可正常测试运行，但提交PTA平台编译错误，不明白~

C++:

#include<iostream>
using namespace std;
int main()
{
    int N;
    cin>>N;
    string stu_id[1001]={""};
    int stu_id2[1001]={0};
    for(int i=0;i<N;i++)
    {
        string str;
        int id1,id2;
        cin>>str>>id1>>id2;
        stu_id[id1]=str;
        stu_id2[id1]=id2;
    }
    int M;
    cin>>M;
    for(int i=0;i<M;i++)
    {
        int id1;
        cin>>id1;
        cout<<stu_id[id1]<<" "<<stu_id2[id1]<<endl;
    }
    return 0;
}