PAT甲级1080 Graduate Admission【模拟】

题目：https://pintia.cn/problem-sets/994805342720868352/problems/994805387268571136

题意：

模拟高考志愿录取。

考生根据总成绩和高考成绩排名。根据排名往下录取，每个人有k个志愿。

如果他填的学校名额没有满，那么就可以被录取。如果他填的学校名额满了，最后一名的排名和这个人是一样的话，就可以被录取。

思路：

直接模拟。

刚开始居然有一个数据超时了，把排序的cmp里面改成传引用居然就过了。

//#include<bits/stdc++>
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<stdlib.h>
#include<queue> 
#include<map>
#include<stack>
#include<set>

#define LL long long
#define ull unsigned long long
#define inf 0x3f3f3f3f 

using namespace std;

int n, m, k;
const int maxn = 4e4 + 5;
struct student{
    int ge, gi;
    int id; 
    int gfinal;
    vector<int>apply;
}stu[maxn];

bool cmp(student &a, student &b)
{
    if(a.gfinal == b.gfinal){
        return a.ge > b.ge;
    }
    else return a.gfinal > b.gfinal;
}

struct school{
    int quota;
    vector<student>addmit;
}sch[105]; 

bool cmpans(student &a, student &b)
{
    return a.id < b.id;
}


int main()
{
    scanf("%d%d%d", &n, &m, &k);
    for(int i = 0; i < m; i++){
        scanf("%d", &sch[i].quota);
    }
    for(int i = 0; i < n; i++){
        stu[i].id = i;
        scanf("%d%d%", &stu[i].ge, &stu[i].gi);
        stu[i].gfinal = (stu[i].ge + stu[i].gi) / 2;
        for(int j = 0; j < k; j++){
            int s;
            scanf("%d", &s);
            stu[i].apply.push_back(s);
        }
    }
    sort(stu, stu + n, cmp);
    
    for(int i = 0; i < n; i++){
        for(int j = 0; j < k; j++){
            int s = stu[i].apply[j];
            if(sch[s].quota){
                sch[s].addmit.push_back(stu[i]);
                sch[s].quota--;
                break;
            }
            else{
                if(sch[s].addmit.size()){
                    student lastone = sch[s].addmit.back();
                    if(lastone.gfinal == stu[i].gfinal && lastone.ge == stu[i].ge){
                        sch[s].addmit.push_back(stu[i]);
                        break;
                    }
                }
            }
        }
    }
    
    for(int i = 0; i < m; i++){
        if(sch[i].addmit.size()){
            sort(sch[i].addmit.begin(), sch[i].addmit.end(), cmpans);
            printf("%d", sch[i].addmit[0].id);
            for(int j = 1; j < sch[i].addmit.size(); j++){
                printf(" %d", sch[i].addmit[j].id);
            }    
        }
        printf("
");
    }
    
    
    return 0;    
}