PAT A 1080 AC代码

测试点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",&quota[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;
}