1080. Graduate Admission (30)

最新推荐文章于 2024-11-21 19:33:30 发布

原创最新推荐文章于 2024-11-21 19:33:30 发布 · 617 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #PAT #Graduate Admission

PAT(Advanced Level) 专栏收录该内容

103 篇文章

订阅专栏

本文介绍了一种处理学校录取问题的算法，通过为每所学校分配额外变量记录最后被录取学生的排名，并以此来判断后续相同排名的学生是否可以被录取。该算法使用C++实现，包括学生和学校的数据结构定义及主要逻辑。

在处理超出学校名额的情况仍然有可能录取的情况时，为每个学校加入一个变量以指示最后一个录入同学的排名，当后来的同学与这个排名相同时则不再考虑是否名额不够。

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>

using namespace std;

struct School{
	int quota;
	int rank;  // the last one's rank
	vector<int> admits;

	School() : quota(0), rank(-1){}
};

struct Student{
	int id, ge, gi, final, rank;
	vector<int> choices;

	Student(int id, int ge, int gi, vector<int> c) : id(id), ge(ge), gi(gi), final((ge+gi)/2), choices(c){}

	bool operator < (const Student& rhs) const{
		if(final != rhs.final) return final > rhs.final;
		else return ge > rhs.ge;
	}
};

int main(){
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k);

	vector<School> schools(m);
	for(int i = 0; i < m; ++i){
		scanf("%d", &schools[i].quota);
	}

	vector<Student> stus;
	for(int i = 0; i < n; ++i){
		int ge, gi;
		vector<int> c(k);

		scanf("%d%d", &ge, &gi);
		for(int j = 0; j < k; ++j){
			scanf("%d", &c[j]);
		}

		stus.emplace_back(i, ge, gi, c);
	}

	sort(begin(stus), end(stus));

	for(size_t i = 0; i < stus.size(); ++i){
		if(i == 0) stus[i].rank = 1;
		else{
			if(stus[i].final == stus[i-1].final && stus[i].ge == stus[i-1].ge){
				stus[i].rank = stus[i-1].rank;
			}else{
				stus[i].rank = i+1;
			}
		}
	}

	for(auto& s : stus){
		for(auto& c : s.choices){
			if(schools[c].quota > 0 || schools[c].rank == s.rank){
				schools[c].admits.push_back(s.id);
				schools[c].quota--;
				schools[c].rank = s.rank;
				break;
			}
		}
	}

	for(auto& s : schools){
		sort(begin(s.admits), end(s.admits));

		bool first = true;
		for(auto& a : s.admits){
			if(first) first = false;
			else printf(" ");
			printf("%d", a);
		}
		printf("\n");
	}



	return 0;
}