1. 程式人生 > >PTA題目----列印學生選課清單

PTA題目----列印學生選課清單

假設全校有最多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

讀完題目第一個想到的就是鍵值對...所以需要用到雜湊表

解題程式碼:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
int mhash(char * name) {
	return (name[0] - 'A') * 26 * 26 * 10 + (name[1] - 'A') * 26 * 10 + (name[2] - 'A') * 10 + name[3] - '0';
}
vector<int> stu[26 * 26 * 26 * 10 + 1];
int main() {
	int N, K;
	scanf("%d %d", &N, &K);
	int i, j;
	char name[5];
	int id, n;
	for(i = 0; i < K; i++) {
		scanf("%d %d", &id, &n);
		for(j = 0; j < n; j++) {
			scanf("%s", name);
			int h = mhash(name);
			stu[h].push_back(id);
		}
	}
	for(i = 0; i < N; i++) {
		scanf("%s", name);
		int h = mhash(name);
		sort(stu[h].begin(), stu[h].end());
		int s = stu[h].size();
		printf("%s %d", name, s);
		for(j = 0; j < s; j++) {
			printf(" %d", stu[h][j]);
		}
		printf("\n");
	}
	return 0;
}

程式碼解析:

給定函式mhash是為了建立一個不會衝突的鍵名,然後使用動態陣列vector建立連結串列

類似於下面這種結構:

另外再給大家一段比較長的程式碼,思路都是一樣的,但是使用不同的資料結構,整體程式碼都會不一樣,大家感受一下(推薦使用上面的那一種)

#include <stdio.h>
#include<malloc.h>
 
typedef struct node {
	int Course;
	struct node *Next;
}Node;
 
Node *Insert(Node *sort, int K);
void Print(Node *head,int N);
 
int main(int argc, char **argv)
{
	Node *list[26][26][26][10];//最後一位放空,避免越界
	int N, K;
	int i, j;
	int snapCourse, stus;
	char snapName[5];
	for (int i = 0; i<26; i++)
		for (int j = 0; j<26; j++)
			for (int u = 0; u<26; u++)
				for (int k = 0; k<10; k++)
					list[i][j][u][k] = NULL;
	scanf("%d %d", &K, &N);
	
	for (i = 0; i < N; i++)
	{
		scanf("%d %d", &snapCourse, &stus);
		for (j = 0; j < stus; j++)
		{
			getchar();
			scanf("%s", snapName);
			list[snapName[0] - 'A'][snapName[1] - 'A'][snapName[2] - 'A'][snapName[3] - '0'] =
				Insert(list[snapName[0] - 'A'][snapName[1] - 'A'][snapName[2] - 'A'][snapName[3] - '0'], snapCourse);
		}
	}
	for (i = 0; i < K; i++)
	{
		scanf("%s", snapName);
		printf("%s", snapName);
		Print(list[snapName[0] - 'A'][snapName[1] - 'A'][snapName[2] - 'A'][snapName[3] - '0'], 0);
		putchar('\n');
	}
 
	return 0;
}
Node *Insert(Node *sort, int K)
{
	if (!sort || sort->Course < K)
	{
		Node *temp = (Node*)malloc(sizeof(Node));
		temp->Course = K;
		temp->Next = sort;
		return temp;
	}
	else if (sort->Course != K)
		sort->Next = Insert(sort->Next, K);
	return sort;
}
void Print(Node *head, int N)
{
	if (head)
	{
		Print(head->Next, ++N );
		printf(" %d", head->Course);
	}
	else 
		printf(" %d", N);
}

最後附上此題目的一個簡單版,輸入相同,但在輸出時,直接程式排序輸出課表: