PTA练习——7-9 天梯赛座位分配

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

注意点：数组大于等于10^6级别时放在main函数外面，否则会运行出错（本题数组开10^5即可，为了方便可以都开大一些）；对于逻辑要求高的最好先列出思路，在纸上写一下再敲代码，还有就是细节一定要注意，不要怕出错，今天老师和我们说程序员就是在改BUG中不断成长的。

#include <bits/stdc++.h>

using namespace std;
int a[1000000]={0};
int main()
{
    int i,j,x,k=0,n,m[110],cnt,l,flag=0;
    cin>>n;
    cnt=n;
    for(i=1;i<=n;i++){
        cin>>x;
        m[i]=x*10;
    }
    while(cnt>0){
        for(i=1;i<=n;i++){
            if(m[i]>0){//还有学生未排座
                if(cnt==1){//只剩下一队人
                    if(k==0||flag==0){
                        flag=1;
                        k++;
                        a[k]=i;//第一个不用隔开
                    }
                    else{
                        k+=2;
                        a[k]=i;
                    }

                }
                else{
                    k++;
                    a[k]=i;
                }
                m[i]--;
                if(m[i]==0)
                    cnt--;
            }
        }
    }
    for(i=1;i<=n;i++){
        cout<<'#'<<i<<endl;
        l=0;
        for(j=1;j<=k;j++){
            if(a[j]==i){//如果第j个下标存的是i号学校输出
                l++;
                if(l%10!=0)cout<<j<<" ";
                else cout<<j<<endl;
            }
        }
    }
    return 0;
}