pta 習題集5-18 列印學生選課清單
阿新 • • 發佈:2018-12-31
假設全校有最多40000名學生和最多2500門課程。現給出每門課的選課學生名單,要求輸出每個前來查詢的學生的選課清單。
輸入格式:
輸入的第一行是兩個正整數:N(≤40000),為前來查詢課表的學生總數;K(≤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
用cin會超時
用Hash ,由於只要3個大寫字母和一個數字,就不要用map了
#include <iostream> #include <string.h> #include <stdlib.h> #include <stdio.h> #include <math.h> #include <string> #include <map> #include <algorithm> #include <vector> using namespace std; int n,k; map<string,int> m; char s[10]; vector<int> a[200005]; int fun(string s) { return ((s[0]-'A')*26*26+(s[1]-'A')*26+(s[2]-'A'))*10+s[3]-'0'; } int main() { scanf("%d%d",&n,&k); int x; int cnt=0; int y; for(int i=1;i<=k;i++) { scanf("%d",&y); scanf("%d",&x); for(int j=1;j<=x;j++) { scanf("%s",s); a[fun(s)].push_back(y); } } for(int i=1;i<=n;i++) { scanf("%s",s); printf("%s",s); int x=fun(s); int len=a[x].size(); printf(" %d",len); sort(a[x].begin(),a[x].end()); for(int i=0;i<len;i++) { printf(" %d",a[x][i]); } printf("\n"); } return 0; }