PAT 乙级 1041 考试座位号

1041 考试座位号
分数 15
作者 CHEN, Yue
单位 浙江大学
每个 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

解题思路:此题就是利用结构体简单循环，python就是用列表。。。

#include<stdio.h>
typedef struct stu
{
	long long id;
	int shi;
	int zheng;
}stu;
int main()
{
	int n;
	scanf("%d",&n);
	stu s[n];
	int i;
	for(i=0;i<n;i++) 
	scanf("%lld %d %d",&s[i].id,&s[i].shi,&s[i].zheng);
	int k;
	scanf("%d",&k);
	int temp;
	int j;
	for(i=0;i<k;i++)
	{
		scanf("%d",&temp);
		for(j=0;j<n;j++)
		{
			if(s[j].shi==temp)
			{
				printf("%lld %d\n",s[j].id,s[j].zheng);
				break;
			}
		}
	}
	
	return 0;
}

python代码:

n=int(input())
s=[]
for i in range(n):
    s.append(input().split())
m=int(input())
a=input().split()
for i in range(m):
    for j in range(n):
        if s[j][1] == a[i]:
            print(s[j][0],s[j][2])
            break