PTA 5-18 打印学生选课清单 (25分)【】

5-18 打印学生选课清单   (25分)

假设全校有最多40000名学生和最多2500门课程。现给出每门课的选课学生名单，要求输出每个前来查询的学生的选课清单。

输入格式:

输入的第一行是两个正整数：N（$\le$40000），为前来查询课表的学生总数；K（$\le$2500），为总课程数。此后顺序给出课程1到K的选课学生名单。格式为：对每一门课，首先在一行中输出课程编号（简单起见，课程从1到K编号）和选课学生总数（之间用空格分隔），之后在第二行给出学生名单，相邻两个学生名字用1个空格分隔。学生姓名由3个大写英文字母+1位数字组成。选课信息之后，在一行内给出了N个前来查询课表的学生的名字，相邻两个学生名字用1个空格分隔。

输出格式:

对每位前来查询课表的学生，首先输出其名字，随后在同一行中输出一个正整数C，代表该生所选的课程门数，随后按递增顺序输出C个课程的编号。相邻数据用1个空格分隔，注意行末不能输出多余空格。

输入样例:

10 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
ZOE1 ANN0 BOB5 JOE4 JAY9 FRA8 DON2 AMY7 KAT3 LOR6

输出样例:

ZOE1 2 4 5
ANN0 3 1 2 5
BOB5 5 1 2 3 4 5
JOE4 1 2
JAY9 4 1 2 4 5
FRA8 3 2 4 5
DON2 2 4 5
AMY7 1 5
KAT3 3 2 4 5
LOR6 4 1 2 4 5

23分版。。。。

#include<iostream>
#include<cstdio>
#include<math.h>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<list>
#include<sstream>
#include<algorithm>
using namespace std;
#define rep(i,j,k)for(i=j;i<k;i++)
#define per(i,j,k)for(i=j;i>k;i--)
#define MS(x,y)memset(x,y,sizeof(x))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll long long
#define abs(x) (x>0?x:-x)
const int INF=0x7ffffff;
const ll MAX=1e18;

const int M=4e4;
int i,j,k,n,m;
vector<int>num[M];
map<string,int>name;
vector<int>::iterator it;

int main()
{
    while(cin>>n>>m){
        int sum=0;
        while(m--){
            int a,b;
            string s;
            cin>>a>>b;
            for(i=0;i<b;i++){
                cin>>s;
                if(!name.count(s)){
                    num[sum].push_back(a);
                    name[s]=sum;sum++;}
                else num[name[s]].push_back(a);
                }
          }
          while(n--){
            string ss;
            cin>>ss;
            cout<<ss<<" "<<num[name[ss]].size()<<" ";
            sort(num[name[ss]].begin(),num[name[ss]].end());
            for(i=0;i<num[name[ss]].size();i++)
                if(!i)cout<<num[name[ss]][i];
                else cout<<" "<<num[name[ss]][i];
            cout<<endl;
        }
    }
    return 0;
}

当时用的结构体。。。超内存啦，，，都忘了查看提示。。。

#include<iostream>
#include<cstdio>
#include<math.h>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<list>
#include<sstream>
#include<algorithm>
using namespace std;
#define rep(i,j,k)for(i=j;i<k;i++)
#define per(i,j,k)for(i=j;i>k;i--)
#define MS(x,y)memset(x,y,sizeof(x))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll long long
#define abs(x) (x>0?x:-x)
const int INF=0x7ffffff;
const ll MAX=1e18;

const int M=1e4+10;
struct node
{
    string name;
    int num[M];
    int sum;
}p[M];
int i,j,k,n,m;

int main()
{
    while(cin>>n>>m){
        int ans=0;
        for(i=0;i<n;i++){
          MS(p[i].num,0);
          p[i].sum=0;
        }
        while(m--){
            int a,b;
            string s;
            cin>>a>>b;
            for(i=0;i<b;i++){
                cin>>s;
                if(!ans){
                    p[0].name=s;p[0].num[0]=a;
                    p[0].sum++;ans++;
                }
                else{
                    int flag=0;
                    for(j=0;j<ans;j++){
                        if(p[j].name==s){
                            p[j].num[p[j].sum]=a;
                            p[j].sum++;
                            flag=1;ans++;
                            break;
                        }
                    }
                    if(!flag){
                        p[ans].name=s;p[ans].num[0]=a;
                        p[ans].sum++;
                        ans++;
                    }
                }
            }
        }
        string ss;
        while(n--){
             cin>>ss;
            for(i=0;i<ans;i++){
                if(p[i].name==ss){
                    cout<<ss<<" "<<p[i].sum<<" ";
                    for(j=0;j<p[i].sum-1;j++)
                        cout<<p[i].num[j]<<" ";
                    cout<<p[i].num[p[i].sum-1]<<endl;
                }
            }
        }
    }
    return 0;
}