PAT——1080 Graduate Admission 甲级

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805387268571136

AC代码

第一遍AC

#include<bits/stdc++.h>
using namespace std;
int n,m,k,t;
struct node{
	int id,ge,gi,sum;
};
vector<node> vec;
unordered_map<int,int> num;
unordered_map<int,vector<int>> plan;
bool cmp(node x,node y){
	if(x.sum==y.sum) return x.ge>y.ge;
	else return x.sum>y.sum;
}
int main()
{
	cin>>n>>m>>k;
	for(int i=0;i<m;i++)
	{
		cin>>t;
		num[i]=t;
	}
	for(int i=0;i<n;i++)
	{
		node tmp;
		tmp.id=i;
		cin>>tmp.ge>>tmp.gi;
		tmp.sum=tmp.ge+tmp.gi;
		vec.push_back(tmp);
		for(int j=0;j<k;j++)
		{
			cin>>t;
			plan[i].push_back(t);
		}
	}
	sort(vec.begin(),vec.end(),cmp);
	int cnt=1;
	vector<vector<int>> res(m);
	for(int i=0;i<n;i++)
	{
		if(i!=n-1&&vec[i+1].sum==vec[i].sum&&vec[i+1].ge==vec[i].ge)
		{
			cnt++;
			continue;
		}
		else 
		{
			vector<int> tnum(m);
			for(int j=i;j>i-cnt;j--)
			{
				for(int l=0;l<plan[vec[j].id].size();l++)
				{
					int school = plan[vec[j].id][l];
					if(num[school]>0)
					{
						tnum[school]++;
						res[school].push_back(vec[j].id);
						break;
					}
				}
			}
			for(int j=0;j<m;j++)
				num[j]-=tnum[j];
			cnt=1;
		}
	}
	for(int i=0;i<m;i++)
	{
		sort(res[i].begin(),res[i].end());
		for(int j=0;j<res[i].size();j++)
		{
			if(j!=0) cout<<" ";
			cout<<res[i][j];
		}
		cout<<endl;
	}
}

第二遍AC（参考）

#include<bits/stdc++.h>
using namespace std;
int n,m,k,t;
struct node{
	int id,ge,gi,sum;
};
vector<node> vec;
unordered_map<int,int> num;
unordered_map<int,vector<int>> plan;
bool cmp(node x,node y){
	if(x.sum==y.sum) return x.ge>y.ge;
	else return x.sum>y.sum;
}
bool cmp1(node x,node y){
	return x.id<y.id;
}
int main()
{
	cin>>n>>m>>k;
	for(int i=0;i<m;i++)
	{
		cin>>t;
		num[i]=t;
	}
	for(int i=0;i<n;i++)
	{
		node tmp;
		tmp.id=i;
		cin>>tmp.ge>>tmp.gi;
		tmp.sum=tmp.ge+tmp.gi;
		vec.push_back(tmp);
		for(int j=0;j<k;j++)
		{
			cin>>t;
			plan[i].push_back(t);
		}
	}
	sort(vec.begin(),vec.end(),cmp);
	vector<vector<node>> res(m);
	vector<int> tnum(m);
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<k;j++)
		{
			int school = plan[vec[i].id][j],index=tnum[school]-1;
			if(tnum[school]<num[school]||(index>=0&&vec[i].sum==res[school][index].sum&&vec[i].ge==res[school][index].ge))
			{
				tnum[school]++;
				res[school].push_back(vec[i]);
				break;
			}
		} 
	}
	for(int i=0;i<m;i++)
	{
		sort(res[i].begin(),res[i].end(),cmp1);
		for(int j=0;j<res[i].size();j++)
		{
			if(j!=0) cout<<" ";
			cout<<res[i][j].id;
		}
		cout<<endl;
	}
}

说明

这两段代码的区别就是对应并列排名的处理不同。我原先采用的统计相同排名的个数并记录，然后用循环统一处理；但我在参考文章后，发现可以通过与该学校最近录入的同学作比较，如果总分以及Ge分都一致，说明排名相同，即当前同学也同样可以被录取

参考文章

1080. Graduate Admission (30)-PAT甲级真题