CCF201609-2 火车购票

#include<iostream>
using namespace std;
int main()
{
    int a[20]= {0}; //用来记录每一排所被占的位置
    int n;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        int v;
        cin>>v;
        int flag=1;
        for(int j=0; j<20; j++)
        {
            if(a[j]+v<=5)
            {
                int temp;
                temp=a[j];
                for(int k=0; k<v; k++)
                {
                    cout<<5*j+temp+k+1<<" ";
                    a[j]++;

                }
                flag=0;//当找到之后就不必进行插位了
                break;
            }
        }

        if(flag)//当没有连续位时
        {
            for(int j=0;j<20;j++)
            {
                if(a[j]<5)
                {
                     for(int k=a[j];k<5;k++)
                     {
                         cout<<5*j+k+1<<" ";
                         v--;
                         a[j]++;
                         if(v==0)break;
                     }
                }
                if(v==0)break;
            }
        }
        cout<<endl;
    }
    return 0;

}