L1-049 天梯赛座位分配 (20 分)

题目描述

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

思路

模拟  逐个学校分配座位, 注意特判只有一个学校的情况

代码
 

#include <bits/stdc++.h>
#define PII pair<int, int>
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 500005;
using namespace std;

int n, a[N], id = 1, sum, dif;
map<int, vector<int>> mp; //学校编号->学生座位vector

int main(){
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        a[i] *= 10;
    }
    int x = n;  //未全部分配的学校数
    if(x == 1) dif = 2; //如果一开始只有一个学校, 间隔直接设置为2
    else dif = 1;
    while(x){   //还有学校未分配时继续
        for(int i = 1; i <= n; i++){
            if(!a[i]) continue; //分配完成
            if(--a[i] == 0) x--;
            mp[i].push_back(id);
            id += dif;
            if(x == 1) dif = 2;   //剩余一个学校时, 间隔为2
        }
    }
    for(int i = 1; i <= n; i++){
        cout << "#" << i << endl;
        for(int j = 0; j < mp[i].size(); j++)
            cout << mp[i][j] << ((j + 1) % 10 ? " " : "\n");
    }
    return 0;
}