PAT(甲级)渡劫(十五)-Graduate Admission(30)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xr469786706/article/details/87649638

PAT(甲级)渡劫(十五)-Graduate Admission(30)

代码如下：

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

using namespace std;

const int maxn = 40005;
const int maxschool = 105;

int quota[maxschool];
vector<int> res[maxschool]; // 存储每个学校招收学生的编号
int vis[maxschool][maxn];   // vis[s][i] = 1 表示学校s有录取rank为i的学生

struct Applicant{
    int id;
    int ranks;
    int GE,GI;
    float grades;
    int choose[5];
    bool operator < (const Applicant &tmp)const{
        if(grades == tmp.grades){
            return GE > tmp.GE;
        }else{
            return grades > tmp.grades;
        }
    }
}app[maxn];

int main(){
    //freopen("in.txt","r",stdin);
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i = 0 ; i < m ; i++){// 输入每个学校的额度
        scanf("%d",&quota[i]);
    }
    for(int i = 0 ; i < n ; i++){// 输入每个学生的分数和申报的学校
        scanf("%d%d",&app[i].GE,&app[i].GI);
        for(int j = 0 ; j < k ; j++){
            scanf("%d",&app[i].choose[j]);
        }
        app[i].id = i;
        app[i].grades = (app[i].GE + app[i].GI)/2.0f;
    }
    sort(app,app+n);

    // 排名次
    int cnt = 0;
    for(int i = 0 ; i < n ; i++){
        app[i].ranks = ++cnt;
        for(int j = i+1 ; j < n ; j++){
            if(app[i].grades == app[j].grades && app[i].GE == app[j].GE){
                app[j].ranks = cnt; //与没有排名次的同学比较， 若分数相同，则名次相同
                i++;
            }
        }
    }

    //memset(vis,0,sizeof(vis));  // 初始化所有学校都没有招收名次的学生
    // 分配名额
    for(int i = 0 ; i < n ; i++){
        int j;
        for(j = 0 ; j < k ; j++){
            int s = app[i].choose[j];   // 根据名次高低查看每个同学的各个志向，看对应学校是否还有名额
            if(quota[s] > 0){
                quota[s]--; // 名额减1
                res[s].push_back(app[i].id);    // 学校s确定招收编号为id的学生
                vis[s][app[i].ranks] = 1;
                break;
            }else{
                if(vis[s][app[i].ranks] == 1){
                    // 如果学校招收了同等名次的学生，即使没有名额了，那么该生也可以被录取
                    res[s].push_back(app[i].id);
                    break;
                }
            }
        }
    }
    // 输出每个学校招收学生的编号
    for(int i = 0 ; i < m ; i++){
        int num = res[i].size();
        if(num == 0){
            printf("\n");
        }else{
            sort(res[i].begin(),res[i].end());
            printf("%d",res[i][0]);
            for(int j=1;j<num;j++){
                printf(" %d",res[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}

运行结果：