打印选课学生名单

打印选课学生名单 （25 分）

假设全校有最多40000名学生和最多2500门课程。现给出每个学生的选课清单，要求输出每门课的选课学生名单。

输入格式:

输入的第一行是两个正整数：N（≤40000），为全校学生总数；K（≤2500），为总课程数。此后N行，每行包括一个学生姓名（3个大写英文字母+1位数字）、一个正整数C（≤20）代表该生所选的课程门数、随后是C个课程编号。简单起见，课程从1到K编号。

输出格式:

顺序输出课程1到K的选课学生名单。格式为：对每一门课，首先在一行中输出课程编号和选课学生总数（之间用空格分隔），之后在第二行按字典序输出学生名单，每个学生名字占一行。

输入样例:

10 5
ZOE1 2 4 5
ANN0 3 5 2 1
BOB5 5 3 4 2 1 5
JOE4 1 2
JAY9 4 1 2 5 4
FRA8 3 4 2 5
DON2 2 4 5
AMY7 1 5
KAT3 3 5 4 2
LOR6 4 2 4 1 5

输出样例:

1 4
ANN0
BOB5
JAY9
LOR6
2 7
ANN0
BOB5
FRA8
JAY9
JOE4
KAT3
LOR6
3 1
BOB5
4 7
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1
5 9
AMY7
ANN0
BOB5
DON2
FRA8
JAY9
KAT3
LOR6
ZOE1

用到了c++ stl里的vector，考虑到如果输出时排序的话，需要排很多次，可能会超时，所以输入后先排序。

ps：注意要用scanf和printf，用cin，cout会超时 。

#include<bits/stdc++.h>
using namespace std;
struct node{
	char name[4];
	vector<int> a;
};
bool cmp(struct node a,struct node b)
{
	return strcmp(a.name,b.name)>0?0:1;
}
struct node str[40001];
vector<char*> s[2501];
int main()
{
	int n,k,j,N,i,x;
	scanf("%d%d",&n,&k);
	for(i=0;i<n;i++)
	{
		scanf("%s",&str[i].name);
		scanf("%d",&N);
		for(j=0;j<N;j++)
		{
			scanf("%d",&x);
			str[i].a.push_back(x);
		}
	}
	sort(str,str+n,cmp);
	for(i=0;i<n;i++)
		for(j=0;j<str[i].a.size();j++)
			s[str[i].a[j]].push_back(str[i].name);
	for(i=1;i<=k;i++)
	{
		printf("%d %d\n",i,s[i].size());
		for(j=0;j<s[i].size();j++)
			printf("%s\n",s[i][j]);
	}
}