题目描述
天梯赛每年有大量参赛队员,要保证同一所学校的所有队员都不能相邻,分配座位就成为一件比较麻烦的事情。为此我们制定如下策略:假设某赛场有 N 所学校参赛,第 i 所学校有 M[i] 支队伍,每队 10 位参赛选手。令每校选手排成一列纵队,第 i+1 队的选手排在第 i 队选手之后。从第 1 所学校开始,各校的第 1 位队员顺次入座,然后是各校的第 2 位队员…… 以此类推。如果最后只剩下 1 所学校的队伍还没有分配座位,则需要安排他们的队员隔位就坐。本题就要求你编写程序,自动为各校生成队员的座位号,从 1 开始编号。
输入格式:
输入在一行中给出参赛的高校数 N (不超过100的正整数);第二行给出 N 个不超过10的正整数,其中第 i 个数对应第 i 所高校的参赛队伍数,数字间以空格分隔。
输出格式:
从第 1 所高校的第 1 支队伍开始,顺次输出队员的座位号。每队占一行,座位号间以 1 个空格分隔,行首尾不得有多余空格。另外,每所高校的第一行按“#X”输出该校的编号X,从 1 开始。
思路
模拟 逐个学校分配座位, 注意特判只有一个学校的情况
代码
#include <bits/stdc++.h>
#define PII pair<int, int>
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 500005;
using namespace std;
int n, a[N], id = 1, sum, dif;
map<int, vector<int>> mp; //学校编号->学生座位vector
int main(){
cin >> n;
for(int i = 1; i <= n; i++){
cin >> a[i];
a[i] *= 10;
}
int x = n; //未全部分配的学校数
if(x == 1) dif = 2; //如果一开始只有一个学校, 间隔直接设置为2
else dif = 1;
while(x){ //还有学校未分配时继续
for(int i = 1; i <= n; i++){
if(!a[i]) continue; //分配完成
if(--a[i] == 0) x--;
mp[i].push_back(id);
id += dif;
if(x == 1) dif = 2; //剩余一个学校时, 间隔为2
}
}
for(int i = 1; i <= n; i++){
cout << "#" << i << endl;
for(int j = 0; j < mp[i].size(); j++)
cout << mp[i][j] << ((j + 1) % 10 ? " " : "\n");
}
return 0;
}