结构体排序以及一些特殊情况的考虑。用两个截止数组分别表示最后一位入取到该学校的两个分数值。
如果这次遍历与上一位的rank相同,则判断该两门成绩是否与该学校的两个截止数组相同,如果相同,无聊是否满额,都继续招人。
AC代码:
#include<iostream>
#include<map>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<list>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 999999999
using namespace std;
int sch[100];
struct node
{
int id;
int ge;
int gi;
vector<int> v;
};
set<int> zhao[100];
int jie1[100];
int jie2[100];
node t[40000];
bool cmp(node x,node y)
{
if(x.ge+x.gi!=y.ge+y.gi)
return x.ge+x.gi>y.ge+y.gi;
else
{
return x.ge>y.ge;
}
}
int main()
{
int n,m,k;
scanf("%d %d %d",&n,&m,&k);
for(int i=0;i<m;i++)
{
scanf("%d",&sch[i]);
}
for(int i=0;i<n;i++)
{
t[i].id=i;
scanf("%d %d",&t[i].ge,&t[i].gi);
for(int j=0;j<k;j++)
{
int d;
scanf("%d",&d);
t[i].v.push_back(d);
}
}
sort(t,t+n,cmp);
int uu=t[0].v.front();
zhao[uu].insert(t[0].id);
sch[uu]--;
jie1[uu]=t[0].ge;
jie2[uu]=t[0].gi;
for(int i=1;i<n;i++)
{
int flag=0;
if(t[i].ge==t[i-1].ge&&t[i].gi==t[i-1].gi)
{
flag=1;
}
if(flag==0)
{
for(int j=0;j<t[i].v.size();j++)
{
int tmp=t[i].v[j];
if(sch[tmp]>0)
{
zhao[tmp].insert(t[i].id);
jie1[tmp]=t[i].ge;
jie2[tmp]=t[i].gi;
sch[tmp]--;
break;
}
}
}
else
{
for(int j=0;j<t[i].v.size();j++)
{
int tmp=t[i].v[j];
if(jie1[tmp]==t[i].ge&&jie2[tmp]==t[i].gi)
{
zhao[tmp].insert(t[i].id);
sch[tmp]--;
break;
}
else
{
if(sch[tmp]>0)
{
zhao[tmp].insert(t[i].id);
sch[tmp]--;
break;
}
}
}
}
}
for(int i=0;i<m;i++)
{
if(zhao[i].empty())
{
printf("\n");
}
else
{
set<int>::iterator it=zhao[i].begin();
printf("%d",*it);
it++;
for(;it!=zhao[i].end();it++)
{
printf(" %d",*it);
}
printf("\n");
}
}
}