【题意】
一个模仿考试排名和志愿录取的流程。
共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", "as[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;
}