PAT (Advanced Level) 1080. Graduate Admission (30)

简单题。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<stack>
#include<queue>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;

struct X
{
    int id;
    int ge,gi;
    int ch[];
}s[+];
int n,m,k;
int sch[];
vector<int>ans[];
int le[],li[];

bool cmp(const X&a,const X&b)
{
    if(a.ge+a.gi==b.ge+b.gi) return a.ge>b.ge;
    return a.ge+a.gi>b.ge+b.gi;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=;i<=m;i++) scanf("%d",&sch[i]);
    for(int i=;i<=n;i++)
    {
        s[i].id=i-;
        scanf("%d%d",&s[i].ge,&s[i].gi);
        for(int j=;j<=k;j++)
        {
            scanf("%d",&s[i].ch[j]);
            s[i].ch[j]++;
        }
    }
    sort(s+,s++n,cmp);

    for(int i=;i<=n;i++)
    {
        for(int j=;j<=k;j++)
        {
            if(sch[s[i].ch[j]]>)
            {
                sch[s[i].ch[j]]--;
                ans[s[i].ch[j]].push_back(s[i].id);
                le[s[i].ch[j]]=s[i].ge;
                li[s[i].ch[j]]=s[i].gi;
                break;
            }
            if(le[s[i].ch[j]]==s[i].ge&&li[s[i].ch[j]]==s[i].gi)
            {
                sch[s[i].ch[j]]--;
                ans[s[i].ch[j]].push_back(s[i].id);
                le[s[i].ch[j]]=s[i].ge;
                li[s[i].ch[j]]=s[i].gi;
                break;
            }
        }
    }

    for(int i=;i<=m;i++)
    {
        sort(ans[i].begin(),ans[i].end());
        for(int j=;j<ans[i].size();j++)
        {
            printf("%d",ans[i][j]);
            if(j<ans[i].size()-) printf(" ");
        }
        printf("\n");
    }
    return ;
}