【PAT】1080. Graduate Admission

关键：主要是理清题意，注意这里要标记好很多条件，比如学生被录取后必须标记已被录取，或用其他方法，刚开始时因为sort的比较函数多了一个判断结果超时了

#define LOCAL
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

struct Person{
    int ge,gi,grade;
    int choice[7];
    int id;
    int r;
    bool admit;
}stu[40010];
bool cmp(Person a,Person b){
    if(a.grade!=b.grade) return a.grade>b.grade;
    else return a.ge>b.ge;

}

struct Quota{
    int num;
    int cnt;
    int ch[40010];
}quota[101];
bool cmpId(int a,int b){
    return stu[a].id<stu[b].id;
}
int main()
{
    #ifdef LOCAL
        freopen("data.in","r",stdin);
        freopen("data.out","w",stdout);
    #endif // LOCAL
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);

    for(int i=0;i<m;i++){
        scanf("%d","a[i].num);
        quota[i].cnt=0;
    }
    for(int i=0;i<n;i++){
        scanf("%d%d",&stu[i].ge,&stu[i].gi);
        for(int j=0;j<k;j++)
            scanf("%d",&stu[i].choice[j]);
        stu[i].grade=stu[i].ge+stu[i].gi;
        stu[i].id=i;
        stu[i].admit=false;
    }

    sort(stu,stu+n,cmp);

    stu[0].r=1;
    for(int i=1;i<n;i++){
        if(stu[i].grade!=stu[i-1].grade)
            stu[i].r=i+1;
        else if(stu[i].grade==stu[i-1].grade && stu[i].ge!=stu[i-1].ge)
            stu[i].r=i+1;
        else stu[i].r=stu[i-1].r;
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<k && !stu[i].admit;j++){
            int pos=stu[i].choice[j];
            if(quota[pos].cnt<quota[pos].num){
                quota[pos].ch[quota[pos].cnt++]=i;
                stu[i].admit=true;
            }else if(stu[quota[pos].ch[quota[pos].cnt-1]].r==stu[i].r){
                quota[pos].ch[quota[pos].cnt++]=i;
                stu[i].admit=true;
            }
        }
    }
    for(int i=0;i<m;i++){
        if(quota[i].cnt>0){
            sort(quota[i].ch,quota[i].ch+quota[i].cnt,cmpId);
            for(int k=0;k<quota[i].cnt;k++)
            {
                printf("%d",stu[quota[i].ch[k]].id);

                if(k!=quota[i].cnt-1) printf(" ");
            }
        }

        printf("\n");
    }

    return 0;
}