ccf-2016-09-2 火车购票 100分

问题描述
　　请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
　　假设一节车厢有20排、每一排5个座位。为方便起见，我们用1到100来给所有的座位编号，第一排是1到5号，第二排是6到10号，依次类推，第20排是96到100号。
　　购票时，一个人可能购一张或多张票，最多不超过5张。如果这几张票可以安排在同一排编号相邻的座位，则应该安排在编号最小的相邻座位。否则应该安排在编号最小的几个空座位中（不考虑是否相邻）。
　　假设初始时车票全部未被购买，现在给了一些购票指令，请你处理这些指令。

#include <iostream>
using namespace std;
int a[21][6]={0};
int main()
{
    int n;
    cin>>n;
   for(int s=0;s<n;s++){
        int count;
        cin>>count;
        int flag=0;
        for(int i=0;i<20;i++)
        {
            int j=0;
            while(a[i][j]!=0)
            {
                j++;
            }
           // cout<<"j="<<j<<" ";
            if(count<=5-j)
            {
                for(int k=0;k<count;k++)
                {
                    cout<<i*5+j+k+1<<" ";
                    a[i][j+k]=1;
                }
                flag=1;//买到票且座位号在同一排
                break;
            }
        }
        //需考虑到不能安排到同一排时的情况
        if(flag==0)
        {
            int index=0;
            for(int i=0;i<20;i++)
            {
                for(int j=0;j<5;j++)
                {
                    if(a[i][j]==0)
                    {
                      //  cout<<"i="<<i;
                        cout<<i*5+j+1<<" ";
                        index++;
                        a[i][j]=1;
                    }
                    if(index>=count)
                        break;
                }
            }
        }
        flag=0;
        cout<<endl;
    }
    return 0;
}