PAT天梯赛 L1-049 天梯赛座位分配

题目链接：点击打开链接

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

AC代码：

#include<iostream>
#include<queue>
#include<algorithm>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<cstdio>
#include<cstdlib>
#include<cctype>
#include<cstring>
using namespace std;
const int MAX = 110;
const int INF = 0X3f3f3f;

vector<int> ans[MAX];//模拟放人
int size[MAX];//每个队的人数
bool vis[MAX];//每个队是否访问过

int main() {
	int n;
	cin >> n;
	for(int i = 1; i <= n; i++) {//得到每个队的人数
		cin >> size[i];
		size[i] *= 10;
	}
	int k = 1, num = 0;//k为数 num为循环指标
	while(num < n) {
		for(int i = 1; i <= n; i++) {
			if(ans[i].size() < size[i]) {//人数不满就放进去
				ans[i].push_back(k);
				if(num == n-1) k +=2;//只有剩最后一队 才是累加2
				else k++;
			}

			if(ans[i].size() == size[i] && vis[i] == false) {//人数满一队 num++   前提是之前没满  所以加了vis数组
				vis[i] = true;
				num++;
			}
		}
	}
	int sum;
	for(int i = 1; i <= n; i++) {//输出
		printf("#%d\n", i);
		sum = 1;
		for(int j = 0; j < ans[i].size(); j++) {
			if(sum != 0 && sum % 10 == 0 || j == ans[i].size() - 1)
				cout << ans[i][j] << endl;
			else
				cout << ans[i][j] << " ";
			sum++;
		}
	}
	return 0;
}