PAT-A1047. Student List for Course (25)

VECTOR

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <vector> 
#include <algorithm>
using namespace std;

#define MAX 40000+10
char name[MAX][5];

vector<int> course[2500];

bool cmp(int a, int b)
{
	return strcmp(name[a], name[b]) < 0;
}

int main()
{
	//freopen("in.txt", "r", stdin);
	//freopen("out.txt", "w", stdout);
	int N = 0, K = 0;
	scanf("%d %d", &N, &K);
	for (int i= 0; i < N; i++)
	{
		int t = 0;
		scanf("%s %d", name[i], &t);
		for (int j = 0; j < t; j++)
		{
			int c;
			scanf("%d", &c);
			course[c].push_back(i);
		}
	}
	
	for (int i = 1; i <= K; i++)
	{
		printf("%d %d\n", i, course[i].size());
		sort(course[i].begin(), course[i].end(), cmp);
		for (int j = 0; j < course[i].size(); j++)
		{
			printf("%s\n", name[course[i][j]]);
		}
	}
	return 0;
	
}