PTA天梯赛座位分配

作者 陈越

单位 浙江大学

天梯赛每年有大量参赛队员，要保证同一所学校的所有队员都不能相邻，分配座位就成为一件比较麻烦的事情。为此我们制定如下策略：假设某赛场有 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 开始。

输入样例：

3
3 4 2

输出样例：

#1
1 4 7 10 13 16 19 22 25 28
31 34 37 40 43 46 49 52 55 58
61 63 65 67 69 71 73 75 77 79
#2
2 5 8 11 14 17 20 23 26 29
32 35 38 41 44 47 50 53 56 59
62 64 66 68 70 72 74 76 78 80
82 84 86 88 90 92 94 96 98 100
#3
3 6 9 12 15 18 21 24 27 30
33 36 39 42 45 48 51 54 57 60

思路：

如果第二多的队伍排在最多的队伍前面。

最多的队伍就要从前一个人位置+2开始排一列，输出样例的#2的82就是这一种。

否则就从前一个人位置+1开始排一列，举例如下：

可以看到#2的第4排从81开始。

方法：

用vector存每个学校的位置，判断完队伍最多的学校的特殊点就结束了。

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define endl "\n"

const ll N = 1e3+7;
ll n;
ll d[N];//存每个学校有多少个队伍 
vector<ll>v[N];//存每个学校的位置 

void solve(){
	cin >> n;
	ll mx=0,w1=-1,mx2=0,w2=1e9;
	for(ll i = 1 ; i <= n ; i ++){
		cin >> d[i];
		if(d[i] > mx)mx = d[i] ,w1 = i;//获取队伍个数最多的学校是谁 
	}
	bool flag=0;
	for(ll i = 1 ; i <= n ; i ++)
		if(d[i] != mx && d[i] > mx2)mx2 = d[i] ,w2 = i;//获取队伍个数第二多的学校是谁 
	if(w1 > w2)flag=1;//如果队伍个数最多的在后面,表面 特殊点需要+2 
	ll w=1,x;//当前的位置,下个人需要多少间隔 
	for(ll i = 1 ; i <= mx ; i ++){
		ll f=0,y=10;//还有队伍没排的学校数,一个队伍10个人 
		for(ll j = 1 ; j <= n ; j ++)//遍历还有几个学校还有队伍 
			if(d[j] >= i)f++;
		f > 1 ? x=1 : x=2;
		if(flag && f == 1)//只剩一个学校队伍没有排完 
			w++,flag=0;
		while(y --)//存当前每个学校排的每个位置 
			for(ll j = 1 ; j <= n ; j ++)
				if(d[j] >= i)v[j].push_back(w),w+=x;
	}
	for(ll i = 1 ; i <= n ; i ++){//输出 
		ll f=0;
		cout << "#" << i << endl;
		for(ll j = 0 ; j < v[i].size() ; j ++){
			f++%10 ? cout << " " << v[i][j] : cout << v[i][j] ;
			if(f%10 == 0)cout << endl;
		}
	}
	return;
}

int main(){
	ll t=1;//cin >> t;
	while(t --)solve();
	return 0;
}