PAT A1080 20200120
依旧是排序题目,读懂题目意思,按照排序的优先级排序。
自己解题使用了vector,因为最后的输出,要求按照学生的顺序进行输出,因此需要在遍历完学生的志愿之后,需要对学校招收的学生进行排序。
注意点是:不知道为何在用sort对vector排序的时候,没有加上cmp_1的函数 居然会报错,还没找到具体的原因。
#include<iostream>
#include<stdlib.h>
#include<queue>
#include<stack>
#include<algorithm>
#include<map>
#include<vector>
#include<cstring>
using namespace std;
typedef struct{
int ge,gi,rank,num;
double gf;
int intend[6];
}Student;
typedef struct{
int admitnum=0;
vector<Student> v;
}School;
bool cmp(Student a,Student b){
if(a.gf!=b.gf)
return a.gf>b.gf;
else if(a.ge!=b.ge)
return a.ge>b.ge;
else
return a.num<b.num;
}
bool cmp_1(Student a,Student b){
return a.num<b.num;
}
Student stu[40010];
int main(){
int N,M,K;//N-applicant M-school K-choice
cin>>N>>M>>K;
School sch[M];
int quota[M];
for(int i=0;i<M;i++)
cin>>quota[i];
for(int i=0;i<N;i++){
cin>>stu[i].ge>>stu[i].gi;
stu[i].num=i;
stu[i].gf=(double)(stu[i].ge+stu[i].gi)/2;
for(int j=0;j<K;j++)
cin>>stu[i].intend[j];
}
sort(stu,stu+N,cmp);
stu[0].rank=1;
for(int i=1;i<N;i++){
stu[i].rank=i+1;
if(stu[i].gf==stu[i-1].gf&&stu[i].ge==stu[i-1].ge)
stu[i].rank=stu[i-1].rank;
}
for(int i=0;i<N;i++){
for(int j=0;j<K;j++){
int sc=stu[i].intend[j];
if(sch[sc].admitnum<quota[sc]||sch[sc].v.back().rank==stu[i].rank){
sch[sc].admitnum++;
sch[sc].v.push_back(stu[i]);
break;
}
}
}
for(int i=0;i<M;i++){
if(sch[i].v.size()==0)
cout<<endl;
else{
sort(sch[i].v.begin(),sch[i].v.end(),cmp_1);
for(int j=0;j<sch[i].v.size();j++){
cout<<sch[i].v[j].num;
if(j!=sch[i].v.size()-1)
cout<<" ";
}
cout<<endl;
}
}
return 0;
}