天梯赛座位分配 PTA

 

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef pair<int,int> PII;
const int N=110;
PII a[N]; //记录每个数可以输出几次 用x,y记录
int b[N][N]; //用于输出
int n;
int main()
{
    cin >> n;
    int mx=0;//标记一个学校最多多少人
    for(int i=0;i<n;i++)
    {
        cin >> a[i].x;
        a[i].x*=10;  //记录每个学校有多少人
        if(a[i].x>mx) mx=a[i].x; //找出最多的队伍，后面就按他来排
    }
    int k=1;
    int kk=0;//标记什么时候只剩一个学校的情况
    for(int i=0;i<mx;i++)
        for(int j=0;j<n;j++)
        {
            if(kk==n-1&&a[j].y<a[j].x) //说明此时只剩一个学校
            {
                b[j][a[j].y]=k;
                k=k+2;
                a[j].y++;
                break; //此循环说明只有一个学校，循环到找到学校即可，break提高效率
            }
            else if(a[j].y<a[j].x) //说明对应的学校还没达到人数最大值
            {
                b[j][a[j].y]=k++;
                a[j].y++;
                if(a[j].y==a[j].x) kk++;
            }
        }
    for(int i=0;i<n;i++)
    {
        cout<<"#"<<i+1<<endl;
        for(int j=0;j<a[i].x;)
        {
            printf("%d",b[i][j]);
            j++;
            if(j%10==0) cout<<endl;
            else cout<<" ";
        }
    }
}