PAT（甲级）1047 Student List for Course (25point(s))

题目

题目链接

思路

题目大意：给出每个人所选的科目，要求按科目输出选择的人有哪些；
先来分析一下，我们肯定需要为每个科目开一个数组，存储有哪些人选择了这个科目，但有个问题，名字是字符型的，如果直接存储字符，比较占内存；那么可不可以为每个人赋予一个编号？可以，因为输入时是按人来输入的，一共有n个人，所以我们可以把人从0 - n-1编号；这样就可以在课程数组中保存数字了；
注意，输出时要按名字的字母序进行排序后再输出；

代码

#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 2600;
vector<vector<int>> course;
char name[40005][5];

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

int main()
{
     course.resize(maxn);
     int n, courseCnt, num, t;//总人数,总的科目数，选择的科目数
     scanf("%d%d", &n, &courseCnt);
     for(int i = 0; i < n; i ++){
          scanf("%s%d", name[i], &num);
          for(int j = 0; j < num; j ++){
               scanf("%d", &t);//读取选择的科目id
               course[t].push_back(i);
          }
     }
     for(int i = 1; i <= courseCnt; 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]]);
          }
     }
     system("pause");
     return 0;
}