CCF-CSP 火車購票

問題描述

試題編號：	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。

這題我只做了90分，暫時還不知道錯在哪裏，有哪位大佬知道的還請指出錯誤，謝謝！

#include <bits/stdc++.h>
using namespace std;

bool arr[20][5];

int main() {
    memset(arr, false, sizeof(arr));
    int n, p, i, j, k;
    cin >> n;
    for (i = 0; i < n; ++i) {
        cin >> p;
        for (j = 0; j < 20; ++j) {
            int remain = 0;
            for (k = 0; k < 5; ++k){
                if (arr[j][k] == false) {
                    ++remain;
                }
            }
            if (remain >= p && remain <= 5) {
                for (k = 0; k < 5; ++k) {
                    if (arr[j][k] == false && p > 0) {
                        arr[j][k] = true;
                        cout << j*5 + k + 1 << " ";
                        --p;
                    }
                }
                break;
            }
        }
        cout << endl;
    }
    return 0;
}