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

循环分配数字，需要注意的是只剩一个学校的情况

#include <bits/stdc++.h>

using namespace std;

typedef struct school
{
    int peo_num;
    int flag;
    vector<int> code;
} SCH;

int main()
{
    int N;
    cin>>N;
    SCH sch[N];
    for(int i=0; i<N; i++)
    {
        cin>>sch[i].peo_num;
        sch[i].peo_num *= 10;
        sch[i].flag = 1;
    }

    //只有一个学校
    if(N == 1)
    {
        int temp = 1;//位置
        for(int i=0; i<sch[0].peo_num; i++)
        {
            sch[0].code.push_back(temp);
            temp += 2;
        }
    }
    else
    {
        int temp = 0;
        int sch_num = N;;
        while(sch_num)
        {
            for(int i=0; i<N; i++)
            {
                //如果没有排完
                if(sch[i].peo_num != sch[i].code.size())
                {
                    //剩下的学校不唯一
                    if(sch_num != 1)
                        temp ++;
                    else
                        temp += 2;
                    sch[i].code.push_back(temp);
                }
                //如果已经排完且没有标记
                else if(sch[i].peo_num == sch[i].code.size()&& sch[i].flag == 1)
                {
                    sch[i].flag = 0;
                    sch_num --;//需安排的学校个数-1
                }
            }
        }
    }

    for(int i=0; i<N; i++)
    {
        printf("#%d\n",i+1);
        for(int j=0; j<sch[i].peo_num; j++)
        {
            if((j+1)%10 == 0)
                cout<<sch[i].code[j]<<endl;
            else
                cout<<sch[i].code[j]<<" ";
        }
    }
    return 0;
}