1039 Course List for Student

1039 Course List for Student

题目大意

根据输入每个课程都有谁参加，统计一些同学都参加了哪些课程

算法思想

• 用map，关键字用姓名，值可以用包含课程的结构体，会方便一些
• 需要动态增加每个人的课程，不然会超出内存
• 最后一个测试点数据很多，需要考虑时间和内存

代码

#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
struct student {
	vector<int>course;
	int count = 0;
};
int main() {
	int n, k, index, m;
	string name;
	cin >> n >> k;
	map<string, student>coll;
	vector<string>qu(n);
	for (int i = 0; i < k; i++) {
		cin >> index >> m;
		for (int j = 0; j < m; j++) {
			cin >> name;
			coll[name].course.push_back(index);//节省空间
			coll[name].count++;
		}
	}
	for (int i = 0; i < n; i++)
		cin >> qu[i];
	for (int i = 0; i < n; i++) {
		cout << qu[i] << " " << coll[qu[i]].count;
		if (coll[qu[i]].count != 0) {//大小不为0，可以照常输出
			sort(coll[qu[i]].course.begin(), coll[qu[i]].course.end());//有可能乱序，需要排序
			for (int j = 0; j < coll[qu[i]].count; j++)
				cout << " " << coll[qu[i]].course[j];
		}
		cout << endl;
	}
	return 0;
}