关键:主要是理清题意,注意这里要标记好很多条件,比如学生被录取后必须标记已被录取,或用其他方法,刚开始时因为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;
}