L2-2 天梯赛的赛场安排
天梯赛使用 OMS 监考系统,需要将参赛队员安排到系统中的虚拟赛场里,并为每个赛场分配一位监考老师。每位监考老师需要联系自己赛场内队员对应的教练们,以便发放比赛账号。为了尽可能减少教练和监考的沟通负担,我们要求赛场的安排满足以下条件:
每位监考老师负责的赛场里,队员人数不得超过赛场规定容量 C;
每位教练需要联系的监考人数尽可能少 —— 这里假设每所参赛学校只有一位负责联系的教练,且每个赛场的监考老师都不相同。
为此我们设计了多轮次排座算法,按照尚未安排赛场的队员人数从大到小的顺序,每一轮对当前未安排的人数最多的学校进行处理。记当前待处理的学校未安排人数为 n:
如果 n≥C,则新开一个赛场,将 C 位队员安排进去。剩下的人继续按人数规模排队,等待下一轮处理;
如果 n<C,则寻找剩余空位数大于等于 n 的编号最小的赛场,将队员安排进去;
如果 n<C,且找不到任何非空的、剩余空位数大于等于 n 的赛场了,则新开一个赛场,将队员安排进去。
由于近年来天梯赛的参赛人数快速增长,2023年超过了 480 所学校 1.6 万人,所以我们必须写个程序来处理赛场安排问题。
输入格式:
输入第一行给出两个正整数 N 和 C,分别为参赛学校数量和每个赛场的规定容量,其中 0<N≤5000,10≤C≤50。随后 N 行,每行给出一个学校的缩写(为长度不超过 6 的非空小写英文字母串)和该校参赛人数(不超过 500 的正整数),其间以空格分隔。题目保证每所学校只有一条记录。
输出格式:
按照输入的顺序,对每一所参赛高校,在一行中输出学校缩写和该校需要联系的监考人数,其间以 1 空格分隔。
最后在一行中输出系统中应该开设多少个赛场。
输入样例
10 30
zju 30
hdu 93
pku 39
hbu 42
sjtu 21
abdu 10
xjtu 36
nnu 15
hnu 168
hsnu 20
输出样例:
zju 1
hdu 4
pku 2
hbu 2
sjtu 1
abdu 1
xjtu 2
nnu 1
hnu 6
hsnu 1
16
在考场上的写法是这样写的,没考虑时间复杂度,不仅当时没写出来,而且题补的时候发现这种方法只能拿17分,会tle
#include <unordered_map>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<string,int> PII;
vector <int> v;//存现在开的不是满的考场
unordered_map <string,int> m;//存一个学校被安排去了多少个考场
vector <PII> s;//存还没排座位的学校
string name[5010];//记录学校出现的顺序输出用
bool cmp(const PII & a,const PII & b){
return a.second<b.second;
}
int main(){
int n,c;
cin>>n>>c;
for(int i=0;i<n;i++){
string str;
int x;
cin>>str>>x;
s.push_back(make_pair(str,x));
m[str]=0;
name[i]=str;
}
sort(s.begin(),s.end(),cmp);
int ma;//记录当前考场剩余空位最大数
int sum=0;//记录单开的考场
while(s.size()){//还有学校没排
if(s.back().second>=c){
sum++;
s.back().second-=c;
m[s.back().first]+=1;
if(s.back().second==0) s.pop_back();
}else {
if(s.back().second>ma||s.size()==0){//剩余坐不下
int res=c-s.back().second;
ma=max(ma,res);
m[s.back().first]++;
s.pop_back();
v.push_back(res);
}else {//放的下
for(int i=0;i<v.size();i++){
int it = v[i];
if(it>=s.back().second) {
v[i]-=s.back().second;
m[s.back().first]++;
s.pop_back();
}
}
for(auto it: v) ma=max(ma,it);
}
}
sort(s.begin(),s.end(),cmp);
}
for(int i=0;i<n;i++){
cout<<name[i]<<" "<<m[name[i]]<<endl;
}
cout<<sum+v.size();
}
上面的代码是因为一个学校人数多的话要判断很多次,而且每次判断后要sort
时间复杂度太高了,但是仔细读题可以发现>=c的人数直接开个新考场,而且一定会坐满,所以可以直接让这一部分人先做好,这样每个学校仅需一判,且不要sort
#include <unordered_map>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<string,int> PII;
vector <int> v;//存现在开的不是满的考场
unordered_map <string,int> m;//存一个学校被安排去了多少个考场
vector <PII> s;//存还没排座位的学校
string name[5010];//记录学校出现的顺序输出用
bool cmp(const PII & a,const PII & b){
return a.second<b.second;
}
int main(){
int n,c;
cin>>n>>c;
int sum=0;//记录单开的考场
for(int i=0;i<n;i++){
string str;
int x;
cin>>str;
scanf("%d",&x);
if(x%c)
s.push_back(make_pair(str,x%c));
m[str]+=x/c;
name[i]=str;
sum+=x/c;
}
sort(s.begin(),s.end(),cmp);
int ma;//记录当前考场剩余空位最大数
while(s.size()){//还有学校没排
if(s.back().second>ma||s.size()==0){//剩余坐不下
int res=c-s.back().second;
ma=max(ma,res);
m[s.back().first]++;
s.pop_back();
v.push_back(res);
}else {//放的下
for(int i=0;i<v.size();i++){
int it = v[i];
if(it>=s.back().second) {
v[i]-=s.back().second;
m[s.back().first]++;
s.pop_back();
}
}
for(auto it: v) ma=max(ma,it);
}
}
for(int i=0;i<n;i++){
cout<<name[i]<<" "<<m[name[i]]<<endl;
}
cout<<sum+v.size();
}