1080 Graduate Admission (30 分)难度: 一般 / 知识点: 模拟 排序

Posted 2021-12-07 辉小歌

https://pintia.cn/problem-sets/994805342720868352/problems/994805387268571136
先总的排序，再枚举每一个学生，对于每一个学生枚举他的所有的志愿。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct node{int id,a,b,sum,s[15];}temp;
vector<node>ve;
vector<node>ans[N];
int cnt[N],n,m,k;
bool cmp(node a,node b)
{
    if(a.sum==b.sum) return a.a>b.a;
    return a.sum>b.sum;
}
bool cmp1(node a,node b)
{
    return a.id<b.id;
}
int main(void)
{
    cin>>n>>m>>k;
    for(int i=0;i<m;i++) cin>>cnt[i];
    for(int i=0;i<n;i++)
    {
        temp.id=i; cin>>temp.a>>temp.b;
        temp.sum=temp.a+temp.b;
        for(int j=0;j<k;j++) cin>>temp.s[j];
        ve.push_back(temp);
    }
    sort(ve.begin(),ve.end(),cmp);
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<k;j++)
        {
            int w=ve[i].s[j];//学校的编号
            int t=ans[w].size();//学校当前收的人数
            if(t<cnt[w]||(ans[w][t-1].sum==ve[i].sum&&ans[w][t-1].a==ve[i].a)) //如果人还没满/和最后一个人的成绩一样
            {
                ans[w].push_back(ve[i]);
                break;
            }
        }
    }
    for(int i=0;i<m;i++)
    {
        sort(ans[i].begin(),ans[i].end(),cmp1);
        for(int j=0;j<ans[i].size();j++) 
        {
            cout<<ans[i][j].id;
            if(j!=ans[i].size()-1) cout<<" ";
        }
        cout<<endl;
    }
    return 0;
}