查找学生信息【九度教程第18题】

题目描述：

输入N个学生的信息，然后进行查询。

输入：

输入的第一行为N，即学生的个数(N<=1000)

接下来的N行包括N个学生的信息，信息格式如下：

01 李江 男 21
02 刘唐 男 23
03 张军 男 19
04 王娜 女 19

然后输入一个M(M<=10000),接下来会有M行，代表M次查询，每行输入一个学号，格式如下：

02
03
01
04

输出：

输出M行，每行包括一个对应于查询的学生的信息。
如果没有对应的学生信息，则输出“No Answer!”

样例输入：

4
01 李江 男 21
02 刘唐 男 23
03 张军 男 19
04 王娜 女 19
5
02
03
01
04
03

样例输出：

02 刘唐 男 23
03 张军 男 19
01 李江 男 21
04 王娜 女 19
03 张军 男 19

来源：2003年清华大学计算机研究生机试真题

注意点

1. 如何处理多行输入数据？
2. 输入学生信息的一行数据以什么数据结构存放？每个字段如何定义类型(字符or整型)
3. 用什么查找方式来处理每个查找请求？

回答一下

1. 同样地，用while(scanf("%d",&n)!=EOF)来控制多行输入，其中n为输入的第一个数：学生总人数
2. 以前面题目的经验知道，对于一行学生信息来讲，用结构体存放比较好。其中注意第一个学号字段，最好定义为字符数组而非整型，因为给的测试用例为0开头的序号，不是整型的标准形式
3. 对于每个查找请求，最简单的思路是都遍历一遍所有的学生信息依次比对，这样的时间复杂度为O(m*n)，由于m、n都很大，达到了千万数量级，所以我们用二分查找来减少复杂度。但是要注意，使用二分查找需要先对学生信息进行排序。

代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct student{
	char no[100]; //用字符数组来定义学号
	char name[10];
	char sex[5];
	int age; 
}; 
student buf[1000]; //buf存放学生信息

bool cmp(student a,student b){// 定义升序排列，注意函数的参数不是int型a、b，而是传入的结构体类型变量 
	return a.no<b.no;
} 

int main(){
	int query_buf[10000]; // 存放每个查找请求的查找结果，存放的查找结果是在排序后buf中的下标
	int n;
	while(scanf("%d",&n)!=EOF){
		for(int i=0;i<n;i++){
			scanf("%s%s%s%d",buf[i].no,buf[i].name,buf[i].sex,&buf[i].age);
		}
		
		sort(buf,buf+n,cmp); //二分查找前需要先排序，这里为升序从小到大排列，用定义的cmp规则
		
		int t; // 查找请求个数
		char x[30]; // 查找的序号
		scanf("%d",&t); // t<10000
		for(int j=0;j<t;j++){
			scanf("%s",x); //输入要查询的学号 
			
			//二分查找 
			int low=0;
			int high=n-1;
			int mid;
			int argOfSorted=-1;
			while(low<=high){
				mid=(low+high)/2;
				if(strcmp(buf[mid].no,x)==0){
					argOfSorted = mid; //返回下标，而不是逻辑序号，所以不用+1
					break;
				}
				if(strcmp(buf[mid].no,x)<0){
					low=mid+1;
				}
				else high=mid-1;
			}
			if(argOfSorted==-1){
				printf("No answer!\n"); 
			}
			else
				query_buf[j]=argOfSorted; //依次把查找请求的结果存放到query_buf中
		}
		
		for(int k=0;k<t;k++){
			printf("%s %s %s %d\n",buf[query_buf[k]].no,buf[query_buf[k]].name,buf[query_buf[k]].sex,buf[query_buf[k]].age);
		}
	}
	return 0;
}

代码中要注意的地方：

1. 使用sort函数时传入自定义的规则，规则定义函数中传入的参数类型要和例子中一致，比如说本例中就传入的参数就必须是结构体类型，然后通过比较该结构体中目标字段返回结果。
2. 因为例子中对于每个查找请求不能直接输入查找结果，所以我们处理的时候只能把这个结果存放到另外一个变量query_buf中，最后一起打印出来结果。