PAT A 1080 Graduate Admission (30 分)

一、思路

构造结构体数组，存储考生成绩及ID，并按照要求排序（Gf大者在前，Gf相等则Ge大者在前），进行择校。

遍历考生的目标院校：
1、若该院校剩余名额，录取。
2、若该院校名额已满，但该考生成绩与该院校最后录取的考生成绩相同，录取。

不要忘记输出时按考生id升序排序QAQ

二、代码

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
typedef struct
{
    int id, Ge, Gi, Gf;
}node;
int cmp( node a, node b )
{
    if( a.Gf != b.Gf )
        return a.Gf > b.Gf;
    else return a.Ge > b.Ge;
}
int main()
{
    int N, M, K;
    scanf("%d %d %d", &N, &M, &K);
    vector<node> rec(N), ans[M];
    vector<int> qouta(M);
    vector<vector<int>> choice(N, vector<int>(K));
    for( int i = 0; i < M; ++i )
        scanf("%d", &qouta[i]);
    for( int i = 0; i < N; ++i )
    {
        scanf("%d %d", &rec[i].Ge, &rec[i].Gi);
        rec[i].Gf = rec[i].Ge + rec[i].Gi;
        rec[i].id = i;
        for( int j = 0; j < K; ++j )
            scanf("%d", &choice[i][j]);
    }
    sort( rec.begin(), rec.end(), cmp );
    for( int i = 0, sch, id; i < N; ++i )
        for( int j = 0, sch; j < K; ++j )
        {
            sch = choice[ rec[i].id ][j];
            if( ans[sch].size() < qouta[sch] || ( ans[sch].back().Gf == rec[i].Gf && ans[sch].back().Ge == rec[i].Ge ) )
            {
                ans[sch].push_back(rec[i]);
                break;
            }
        }
    for( int i = 0; i < M; ++i )
    {
        sort( ans[i].begin(), ans[i].end(), [] ( node a, node b ) { return a.id < b.id; } );
        for( int j = 0; j < ans[i].size(); ++j )
            printf("%s%d", j ? " ":"", ans[i][j].id);
        printf("\n");
    }
}