CCF1609-2火車購票

問題描述
　　請實現一個鐵路購票系統的簡單座位分配算法，來處理一節車廂的座位分配。
　　假設一節車廂有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<cmath>

using namespace std;

int main(){
    int n;              //購票次數 
    int a[105];         //每次購買車票數 
    int seat[20][6];    //共20排，每排5個座位由0/1表示是否售出，第六個值爲當前排剩餘空閒座位數 

//  初始化 
    for( int i = 0; i < 20; i++ ){
        for( int j = 0; j < 5 ; j++ ){
            seat[i][j] = 1;     //初始時座位空閒 
        }
        seat[i][5] = 5;         //空閒座位數初始化爲5 
    } 

//  輸入指令 
    cin >> n;
    for(int i = 0; i < n; i++){
        cin >> a[i];
    }

//  處理指令 
    for(int i = 0; i < n; i++){         //處理第i條指令 
        int flag = 0;
        int count = a[i];                   //值爲0代表沒有連續座位 
        for( int j = 0; j < 20; j++ ){  //查找第j排座位 
            if( seat[j][5] >= a[i] ){
                flag = 1;               //有連續座位 
                seat[j][5] -= a[i];     //空閒座位數更新 
                for( int k = 0; k < 5, count > 0; k++ ){
                    if( seat[j][k] == 1 ){
                        cout << j*5 + k+1 << " ";   //輸出座位號 
                        seat[j][k] = 0; 
                        count--;            //座位已售出 
                    }
                }
                break; 
            }
        }

        if( !flag ){                    //沒有連續空座，從小序號開始出售
            for( int j = 0; j < 20; j++ ){
                if( seat[j][5] > 0 ){                   //如果這一排還有剩餘座位，出售 
                    for( int k = 0; k < 5; k++ ){
                        if( seat[j][k] == 1 ){
                            cout << j*5 + k+1 << " ";   //輸出座位號 
                            seat[j][k] = 0;             //座位已售出 
                            count--;
                        }
                    }
                }
                if( count == 0 ){
                    break;
                }
            }
        }
        cout << endl; 
    }

    return 0; 
}