CCF CSP 201609-2 火車購票

CCF計算機職業資格認證考試題解系列文章爲meelo原創，請務必以連接形式註明本文地址

CCF CSP 201609-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張票，獲得座位一、2。
　　2) 購5張票，獲得座位6至10。
　　3) 購4張票，獲得座位11至14。
　　4) 購2張票，獲得座位三、4。

評測用例規模與約定

　　對於全部評測用例，1 ≤  n ≤ 100，全部購票數量之和不超過100。

解析

用一個vector記錄每一排還剩了多少個座位，初始爲5。

用了兩個子函數讓程序的結構更加清晰。

程序

C++

#include <cstdio>
#include <vector>

using namespace std;

vector<int> seats(20,5);

void printSeats(int row, int num) {
    for(int i=1; i<=num; i++) {
        printf("%d ", row*5+i+5-seats[row]);
    }
}

void requstSeats(int num) {
    for(int i=0; i<20; i++) {
        if(seats[i] >= num) {
            printSeats(i, num);
            seats[i] -= num;
            num = 0;
        }
    }
    int i = 0;
    while(num) {
        if(seats[i] > 0) {
            if(num <= seats[i]) {
                printSeats(i, num);
                num = 0;
                seats[i] -= num;
            }
            else {
                printSeats(i, seats[i]);
                num -= seats[i];
                seats[i] = 0;
            }
        }
        i++;
    }
    printf("\n");
}

int main() {
    int N;
    scanf("%d", &N);
    
    int req;
    for(int n=0; n<N; n++) {
        scanf("%d", &req);
        requstSeats(req);
    }
}