PAT (Advanced Level) Practice 1039 Course List for Student (25分)

简单模拟

#include<bits/stdc++.h>
#define ll long long 
const int mod=1e9+7;
using namespace std;
vector<int> v[400005]; //v[i]表示编号为i的人 选的课程号 
map<string,int> m;
int n,k;
int main()
{
	string s;
	cin>>n>>k;
	int a,b,w=1;
	for(int i=1;i<=k;i++)
	{
		cin>>a>>b;
		while(b--)
		{
			cin>>s;
			if(m[s]==0)
			{
				m[s]=w;
				v[w].push_back(a);
				w++;
			}
			else
				v[m[s]].push_back(a);
		}
	}
	for(int i=1;i<=n;i++)
	{
		cin>>s;
		cout<<s;
		int t=m[s];
		cout<<" "<<v[t].size();
		sort(v[t].begin(),v[t].end());
		for(int j=0;j<v[t].size();j++)
				cout<<" "<<v[t][j];	
		cout<<endl;	
	} 
	return 0;
}