【PAT-A1080】Graduate Admission（排序）

【题意】

一个模仿考试排名和志愿录取的流程。
共N个考生，M所学校，每个考生有K个志愿。先给考生按照平均成绩排名，平均成绩相同就按照笔试成绩排名，还相同的话排名就一致。然后按照排名顺序对每个考试进行录取，每个学校都有录取名额，依次考生的每个志愿，如果当前志愿学校还有名额就录取，否则看下一个学校，如果全部志愿都没能录取，就不能录取。另外，如果多并列的人报了同一所学校，而且这所学校只剩最后一个名额的话，学校即使超额也得把并列的人都录取。

【分析】

• 先排序。建立考生结构体，只要一个总成绩和笔试成绩即可，另外也设置一个变量存排名。然后写cmp函数排序。
• 从头到尾遍历rank，每次判断的时候同时判断所有排名相同的考生。比如用i表示当前排名相同的第1个考生，j指向当前排名相同的最后一个考生，从i开始依次判断，直到当前排名相同的最后一个考生。用一个关于每所学校的hash表，在开始录取排名相同考生的时候，hash表全部置为false，当一个考生被第k所学校录取，就设hash[k] = true，后面排名相同的考生，如果当前判断是否录取的学校已满，就判断其hash值是否为true，若为true则也录取，否则继续判断下一个志愿。另外可以用一个vector数组记录每个学校录取的学生编号。
• 注意点，比如max，memset之类的函数，在本地可能不用加cmath,和cstring之类的头文件也可以，因为很可能已经被包含在了algorithm之类的头文件里，但是在oj上却没有，所以最好还是自己加上。

【代码】

#include <stdio.h>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn = 40010;
const int maxm = 110;
struct Student
{
	int sum, grade;
	int rank, index;
	vector<int> choices;
}stus[maxn];

bool cmp(Student& a, Student& b) {
	if (a.sum != b.sum)
	{
		return a.sum > b.sum;
	}
	else {
		return a.grade > b.grade;
	}
}

int quotas[maxm] = { 0 };
vector<int> admissions[maxm];
int main() {
	// 输入
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k);
	for (int i = 0; i < m; i++)
	{
		scanf("%d", &quotas[i]);
	}
	int wGrade, iGrade;
	for (int i = 0; i < n; i++)
	{
		scanf("%d%d", &wGrade, &iGrade);
		stus[i].index = i;
		stus[i].grade = wGrade;
		stus[i].sum = wGrade + iGrade;
		for (int j = 0; j < k; j++)
		{
			int chioce;
			scanf("%d", &chioce);
			stus[i].choices.push_back(chioce);
		}
	}

	// 排序
	sort(stus, stus + n, cmp);
	stus[0].rank = 1;
	for (int i = 0; i < n; i++)
	{
		if (i != 0) {
			if (stus[i].sum == stus[i - 1].sum && stus[i].grade == stus[i - 1].grade) {
				stus[i].rank = stus[i - 1].rank;
			}
			else
			{
				stus[i].rank = stus[i - 1].rank + 1; 
			}
		}
	}
	int i = 0;
	bool hashTable[maxm] = { false };
	while (i < n)
	{
		for (int j = 0; j < stus[i].choices.size(); j++)
		{
			int choice = stus[i].choices[j];
			if (quotas[choice] > 0) {
				quotas[choice]--;
				hashTable[choice] = true;
				admissions[choice].push_back(stus[i].index);
				break;
			}
		}
		i++;
		while (i != 0 && i < n && stus[i].rank == stus[i - 1].rank )
		{
			for (int j = 0; j < stus[i].choices.size(); j++)
			{
				int choice = stus[i].choices[j];
				if (quotas[choice] > 0) {
					quotas[choice]--;
					hashTable[choice] = true;
					admissions[choice].push_back(stus[i].index);
					break;
				}
				else {
					if (hashTable[choice] == true) {
						admissions[choice].push_back(stus[i].index);
						break;
					}
				}
			}
			i++;
		}
		memset(hashTable, false, sizeof(hashTable));
	}

	for (int i = 0; i < m; i++)
	{
		sort(admissions[i].begin(), admissions[i].end());
		for (int j = 0; j < admissions[i].size(); j++)
		{
			if (j != 0) printf(" ");
			printf("%d", admissions[i][j]);
		}
		printf("\n");
	}

	return 0;
}