ZJU-PAT 1080. Graduate Admission (30) 浙大2014年上机复试第四题

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;

const int maxn=40005;
const int maxm=101;

int n,m,k;
int num[maxm];
int lastRank[maxm];

vector<int> mm[maxm];

struct stu
{
int id;
int GE,GI,fin;
int pre[6];
int rank;
};
stu tt[maxn];

bool cmp(stu a,stu b)
{
if(a.fin!=b.fin)
return a.fin>b.fin;
else
return a.GE>b.GE;
}

void Init()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=0; i<m; i++) scanf("%d",&num[i]);

for(int i=0; i<n; i++)
{
scanf("%d%d",&tt[i].GE,&tt[i].GI);
tt[i].fin=(tt[i].GE+tt[i].GI);
tt[i].id=i;
for(int j=0; j<k; j++)
{
scanf("%d",&tt[i].pre[j]);
}
}

sort(tt,tt+n,cmp);

for(int i=0; i<maxm; i++) mm[i].clear();

tt[0].rank=0;
for(int i=1; i<n; i++)
{
if(tt[i].fin==tt[i-1].fin && tt[i].GE==tt[i-1].GE) tt[i].rank=tt[i-1].rank;
else
tt[i].rank=i;
}
}

void Select()
{
for(int i=0; i<n; i++)
{
int curstuid=tt[i].id;

for(int j=0; j<k; j++)
{
int schoolid=tt[i].pre[j];

if(mm[schoolid].size()<num[schoolid])
{
mm[schoolid].push_back(curstuid);
lastRank[schoolid]=tt[i].rank;   //当前学生的排名 非tt[curstuid].rank
break;
}
else if(mm[schoolid].size()>0)
{
/*int laststu=mm[schoolid].back();  错误同上所述
if(tt[laststu].rank==tt[i].rank)
{
mm[schoolid].push_back(curstuid);
break;
}*/
if(tt[i].rank==lastRank[schoolid])
{
mm[schoolid].push_back(curstuid);
break;
}
}
}
}
}

void output()
{
for(int i=0; i<m; i++)
{
if(mm[i].size()==0)
printf("\n");
else
{
sort(mm[i].begin(),mm[i].end());
for(int j=0; j<mm[i].size(); j++)
{
if(j==0)printf("%d",mm[i][j]);
else printf(" %d",mm[i][j]);
}
printf("\n");
}
}
}

int main()
{
Init();
Select();
output();
return 0;
}