测试点1和测试点2一开始想了半天为啥不过
后来发现是排序后结点顺序以及变了,没法直接根据序号找到对应的成绩。
所以加了一个new_index来记录每个学校的新的下标序列,再对应回去查成绩
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int N,M,K;
int quota[100];
struct node{
int num;
int ge,gi;
double fg;
vector<int> school;
};
vector<node> Node;
bool cmp(const node a,const node b){
if(a.fg==b.fg){
return a.ge>b.ge;
}else{
return a.fg>b.fg;
}
}
int main(){
cin>>N>>M>>K;
Node.resize(N);
vector<int> ans[M];
int tag[N]={0};
for(int i=0;i<M;i++){
scanf("%d","a[i]);
}
for(int i=0;i<N;i++){
Node[i].num=i;
scanf("%d %d",&Node[i].ge,&Node[i].gi);
Node[i].fg=(Node[i].ge+Node[i].gi)*1.0/2;
for(int j=0;j<K;j++){
int x;
scanf("%d",&x);
Node[i].school.push_back(x);
}
}
sort(Node.begin(),Node.end(),cmp);
vector<int> new_index[M];
for(int i=0;i<N;i++){
for(int j=0;j<K;j++){
if(tag[Node[i].num]==1) break;
int sc=Node[i].school[j];
if(quota[sc]>0){
ans[sc].push_back(Node[i].num);
new_index[sc].push_back(i);
quota[sc]--;
tag[Node[i].num]=1;
}else if(Node[i].gi==Node[new_index[sc][new_index[sc].size()-1]].gi&&Node[i].ge==Node[new_index[sc][new_index[sc].size()-1]].ge){
ans[sc].push_back(Node[i].num);
new_index[sc].push_back(i);
tag[Node[i].num]=1;
}
}
}
for(int i=0;i<M;i++){
sort(ans[i].begin(),ans[i].end());
for(auto it=ans[i].begin();it!=ans[i].end();it++){
if(it==ans[i].begin()){
printf("%d",*it);
}else{
printf(" %d",*it);
}
}
cout<<endl;
}
return 0;
}