csp火车购票

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int a[21][6],n,p;
    cin>>n;
    int no=0;
    for(int i=1;i<=20;++i)
        for(int j=1;j<=5;++j)
        a[i][j]=++no;
    while(n--)
    {
        cin>>p;
        if(p==1)
        {
            for(int i=1;i<=20;++i)
                for(int j=1;j<=5;++j)
                    if(a[i][j])
                {
                    cout<<a[i][j]<<endl;
                    a[i][j]=0;
                    i=21;break;
                }
                continue;
        }
        int f=0;
        for(int i=1;i<=20;++i)
        {
            for(int j=1;j<=5-p+1;++j)
            {
                int k;
                for(k=j;k<j+p;++k)
                {
                    if(a[i][k]==0)
                    break;
                }
                if(k==j+p)
                {
                    k-=p;
                    for(;k<j+p;++k)
                    {
                        cout<<a[i][k]<<" ";
                        a[i][k]=0;
                    }
                    i=21;f=1;
                    break;
                }
            }
        }
        if(f==0)
        {
            int t=0;
            for(int i=1;i<=20&&t<p;++i)
            for(int j=1;j<=5&&t<p;++j)
            if(a[i][j])
            {
                cout<<a[i][j]<<" ";
                a[i][j]=0;++t;
            }
        }
        cout<<endl;
    }
    return 0;
}