主要是模拟,创建两个结构体保存学生、学校数据。理清思路就好了
这次的bug在于 cmp函数中,if 以后必须出现else而不是else if,否则会导致没有return
还有如果数组过于复杂,将其赋值给一个临时变量,便于理解和书写,比如代码中的 prefer = stu[i].choice[j]
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct student{
int ge;
int gi;
int fg;
int choice[6];
int range;
int mark;
}stu[40010];
struct school{
int quota;
int number=0;
int mark[40010];
}sch[105];
bool cmp(student a,student b){
if (a.fg!=b.fg) return a.fg>b.fg;
else return a.ge>b.ge;
//这里不要写else if (a.ge!=b.ge) 会导致没有return 运行超时, 以及一个测试点不过
}
bool cmp1(int a, int b){
return a<b;
}
int main(){
int n,m,k;
scanf("%d %d %d",&n,&m,&k);
for (int i=0;i<m;i++){
scanf("%d",&sch[i].quota);
}
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].fg = (stu[i].ge+stu[i].gi)/2;
stu[i].mark=i;
}
sort(stu,stu+n,cmp);
int last[m]={-1};
for (int i=0;i<n;i++){
if (i>0 && stu[i].fg==stu[i-1].fg && stu[i].ge == stu[i-1].ge) stu[i].range = stu[i-1].range;
else stu[i].range=i;
for (int j=0;j<k;j++){
int prefer = stu[i].choice[j];
if (sch[prefer].quota>sch[prefer].number || last[prefer] !=-1 && stu[last[prefer]].range==stu[i].range){
last[prefer] = i;
sch[prefer].mark[sch[prefer].number]=stu[i].mark;
sch[prefer].number++;
break;
}
}
}
for (int i=0;i<m;i++){
sort(sch[i].mark,sch[i].mark+sch[i].number,cmp1);
for (int j=0;j<sch[i].number;j++){
if (j==0 ) printf("%d",sch[i].mark[j]);
else printf(" %d",sch[i].mark[j]);
}
printf("\n");
}
return 0;
}