打印选课学生名单（应用map，set）

假设全校有最多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

代码长度限制

16 KB

时间限制

1000 ms

内存限制

64 MB

注意：

1、运用了map和set，头文件需要<map><set>.。

2、输出set时，需要用：

cpp 复制代码

set<string>::iterator j;
	for (int i = 0; i < k; i++)
	{
		cout << i + 1 << " " << course_num[i + 1] << endl;//输出课程，选课人数
		for (j = course[i + 1].begin(); j != course[i + 1].end(); j++)
		{
			cout << *j << endl;//输出选课学生名字
		}
	}

3、最后数据时运行会超时。

cpp 复制代码

#include<iostream>
#include<map>
#include<set>
using namespace std;
int main()
{
	int n, k;
	cin >> n >> k;
	map<int, set<string>> course;//一对多
	int course_num[2501] = { 0 };//记录每门课人数
	for (int i = 0; i < n; i++)
	{
		string name;
		int num;
		cin >> name >> num;
		for (int j = 0; j < num; j++)
		{
			int b;
			cin >> b;
			course[b].insert(name);//将选课的人插入到课程里
			course_num[b]++;//该门课人数加一
		}
	}
	set<string>::iterator j;
	for (int i = 0; i < k; i++)
	{
		cout << i + 1 << " " << course_num[i + 1] << endl;//输出课程，选课人数
		for (j = course[i + 1].begin(); j != course[i + 1].end(); j++)
		{
			cout << *j << endl;//输出选课学生名字
		}
	}
}