想不明白测试点1和2为什么过不了,逻辑跟参考程序一样,结构用的不一样。求好心人瞄一眼,提点迷津。
题目:https://pintia.cn/problem-sets/994805342720868352/problems/994805387268571136
#include <cstdio>
#include <algorithm>
using namespace std;
struct student{
int GE, GI;
int prefer[6];
int id;
}stu[40010];
bool cmp(student a, student b){
if (a.GE + a.GI != b.GE + b.GI)
return a.GE + a.GI > b.GE + b.GI;
else
return a.GE > b.GE;
}
int school[110][40010];
int main(){
int n, m, k, num[110], quota[110];
scanf("%d%d%d", &n, &m, &k);
for(int i = 0; i < m; i++){
scanf("%d", quota+i);
num[i] = 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].prefer+j);
stu[i].id = i;
}
sort(stu, stu + n, cmp);
for(int i = 0; i < n; i++){
for(int j = 0; j < k; j++){
if(quota[stu[i].prefer[j]] == 0){continue;}
else if(num[stu[i].prefer[j]] < quota[stu[i].prefer[j]]){
school[stu[i].prefer[j]][num[stu[i].prefer[j]]] = stu[i].id;
num[stu[i].prefer[j]]++;
break;
}
else if(stu[i].GI == stu[school[stu[i].prefer[j]][num[stu[i].prefer[j]] - 1]].GI && stu[school[stu[i].prefer[j]][num[stu[i].prefer[j]] - 1]].GE == stu[i].GE){
school[stu[i].prefer[j]][num[stu[i].prefer[j]]] = stu[i].id;
num[stu[i].prefer[j]]++;
break;
}
}
}
for(int i = 0; i < m; i++){
if(num[i] > 0){
sort(school[i], school[i]+num[i]);
for(int j = 0; j < num[i]; j++){
printf("%d", school[i][j]);
if(j < num[i] - 1)
printf(" ");
}
printf("\n");
}
}
return 0;
}
。