PTA 天梯赛座位分配

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;
int a[110][11000];
map<int,int>mp;//map用来标记每个学校有多少队伍；
int n , m , t[105] , sum , seat , flag , ans;//t数组也是用来标记剩余的队伍数；
void cinteam(int a){//这是一个输入队伍数的函数；
    for(int i = 1 ; i <= a ; i ++){
        cin >> t[i];
        mp[i] = t[i];
        sum += t[i];//总队伍数，*10就得到总人数；
    }
}
bool check(){//检测是否剩下队伍只有一个；
    int c = 0;
    for(int i = 1 ; i <= n ; i ++)
        if(t[i])c ++;
    if(c == 1)return true;
    return false;
}
int main(){
    ios::sync_with_stdio(false);
    while(cin >> n){
        m = 1;flag = 0;//m用来表示第几个学校；
        memset(a,0,sizeof(a));
        cinteam(n);
        for(int i = 1 ; i <= sum* 10 ; i ++){//从1号跑到最后；
            if(check()){
                flag = 1;
                ans = i;
                break;
            }//只有一个队伍就退出，flag= 1，下面再进行处理；
            for(int j = 0 ; j < mp[m] * 10 ; j ++){
                if(!a[m][j]){
                    seat = j;
                    break;
                }
            }//找到空的位置直接给seat；
            a[m][seat] = i;
            if(i == sum * 10)break;
            if((seat + 1)% 10 == 0)t[m] --;//如果已经有10个格子有id，那就减去一个队；
            m ++;
            if(!t[m]){
                if(m < n)//判断空学校的后面是否还有学校有队伍；
                    while(m <= n){
                    m ++;
                    if(t[m])break;
                }//如果全空就从前再往后跑；
          		if(!t[m])for(int j = 1 ; j <= n ; j ++){
                    if(t[j] != 0){
                        m = j;
                        break;
                    }
                }
            }
        }
        if(flag){
            for(int i = 1 ; i <= n ; i ++){
                if(t[i]){
                    if(i == 1)seat ++;
                    if(n == 1)seat --;//是第一个的话seat已经跑过去了，得加1，
                    			      //但是如果只有一个学校，那就是从0开始的；
                    for(int j = 1 ; j <= t[i] * 10 ; j ++){
                        a[i][seat] = ans;
                        seat ++;
                        ans += 2;
                    }
                }
            }
        }
        for(int i = 1 ; i <= n ; i ++){
            cout << "#" << i << endl;
            for(int j = 0 ; j < mp[i] * 10 ; j ++){
                if(j % 10 != 0)cout << " ";
                cout << a[i][j];
                if((j + 1)% 10 == 0)cout << "\n";
            }
        }
    }
    return 0;
}

说实话还是有点阴间的这个题目，写了好久。。。。。。

更新后的代码：

差不多一年过去了，学了好多的容器以及一些别的，发现这题当年写的实在是，太丑陋了；
再来讲一下思路吧：
开个数组b记录每个队还有多少人需要被安排座位；
用vector来存储每个人的位置；
用一个变量nowteam来记录还剩几个队要排位置，now来代表现在是第几个队伍；
写一个函数，看哪个队还有人没位置，下一个位置就是哪个队伍的；

int nextteam(int x){
    while(!b[x]){
        x ++ ;
        if(x > n)x = 1;
    }return x;
}

然后是总代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n , T , a[105] , b[105];
vector<int>v[105];
int nextteam(int x){
    while(!b[x]){
        x ++ ;
        if(x > n)x = 1;
    }return x;
}
int main(){
    cin >> n;
    for(int i = 1 ; i <= n ; i ++)cin >> a[i] , b[i] = a[i] * 10;
    int now = 0 , nowseat = 1 , nowteam = n;
    while(nowteam){
        now = nextteam(now + 1);
        b[now] --;
        v[now].push_back(nowseat);
        nowseat += (nowteam == 1 ? 2 : 1);
        if(!b[now])nowteam --;
    }
    for(int i = 1 ; i <= n ; i ++){
        cout << "#" << i << endl;
        for(int j = 0 ; j < v[i].size() ; j ++){
            cout << v[i][j];
            (j+1)%10 ? cout << " " : cout << "\n";
        }
    }
}