201609-2火車購票

問題描述

試題編號：	201609-2
試題名稱：	火車購票
時間限制：	1.0s
內存限制：	256.0MB
問題描述：	問題描述 　　請實現一個鐵路購票系統的簡單座位分配算法，來處理一節車廂的座位分配。 　　假設一節車廂有20排、每一排5個座位。爲方便起見，我們用1到100來給所有的座位編號，第一排是1到5號，第二排是6到10號，依次類推，第20排是96到100號。 　　購票時，一個人可能購一張或多張票，最多不超過5張。如果這幾張票可以安排在同一排編號相鄰的座位，則應該安排在編號最小的相鄰座位。否則應該安排在編號最小的幾個空座位中（不考慮是否相鄰）。 　　假設初始時車票全部未被購買，現在給了一些購票指令，請你處理這些指令。 輸入格式 　　輸入的第一行包含一個整數n，表示購票指令的數量。 　　第二行包含n個整數，每個整數p在1到5之間，表示要購入的票數，相鄰的兩個數之間使用一個空格分隔。 輸出格式 　　輸出n行，每行對應一條指令的處理結果。 　　對於購票指令p，輸出p張車票的編號，按從小到大排序。 樣例輸入 4 2 5 4 2 樣例輸出 1 2 6 7 8 9 10 11 12 13 14 3 4 樣例說明 　　1) 購2張票，得到座位1、2。 　　2) 購5張票，得到座位6至10。 　　3) 購4張票，得到座位11至14。 　　4) 購2張票，得到座位3、4。 評測用例規模與約定 　　對於所有評測用例，1 ≤ n ≤ 100，所有購票數量之和不超過100。

#include <iostream>
#include<memory.h>

#define N 100
#define ROW 20
using namespace std;

/*從line排開始尋找到line+count-1排 ，尋找到座位數大於等於num的排*/
int findseat(int *line ,int count ,int num)
{
    for(int i=1 ;i<=count ;i++)
    {
        if(5-(*line)>=num)
            return i;
        line++;
    }

    return 0;
}


int main()
{
    int n ,row ,start;
    int line[ROW+1] ,instr[N] ;

    memset(line ,0 ,(ROW+1)*sizeof(int));

    cin>>n;
    for(int i=0 ;i<n ;i++)
    {
        cin>>instr[i];
    }

    for(int i=0 ;i<n ;i++)
    {
        row=findseat(line+1 ,ROW ,instr[i]);
        if( row != 0)
        {
            start = line[row]+1;
            while(instr[i]--)
            {
                cout<<5*(row-1)+start<<' ';
                start++;
                line[row]++;
            }
            cout<<endl;
        }
        else
        {
            for(int j=0;j<instr[i] ; )
            {
                row = findseat(line+1 ,ROW ,1);

                while(line[row]<5&&j<instr[i])
                {
                    j++;
                    line[row]++;
                    cout<<5*(row-1)+line[row]<<' ';
                }
            }
            cout<<endl;
        }
    }
    return 0;
}