問題描述
請實現一個鐵路購票系統的簡單座位分配算法,來處理一節車廂的座位分配。
假設一節車廂有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。
策略是按順序優先分配連號座位,再逐一插空單個座位,這樣能保證座位最大利用且連號的人坐一起。對於一排來說,只存在5個座位,並且分配方式是從左到右,因此只需要記錄每排剩餘座位數量,而不需要記錄具體坐了幾排幾號。
在分配時,首先整排分配,因爲一人購票不超過5張,因此最多一次分配一排,所有整排分配確定後,再將零星少於5個人的位置逐排判斷是否有足夠容量。
在號數分配的處理上,直接通過排號與當前排剩餘座位數計算,確定分配號數的左邊界與右邊界,在確定號數邊界後即可通過遍歷直接輸出分配座位號結果。
#include<stdio.h>
#include<string.h>
using namespace std;
const int maxn = 108;
int cnt[maxn];
int main()
{
int n,p;
while (scanf("%d",&n)!=EOF)
{
for(int i = 0; i < n ; i++) cnt[i] = 5;
for (int i = 0; i < n; i++)
{
scanf("%d",&p);
for (int j = 0; j < 20; j++)
{
if(cnt[j] >= p)
{
int l = j * 5 + 6 - cnt[j];
int r = l + p;
for(int k = l; k < r; k++)
printf("%d%c", k, k==r-1?'\n':' ');
cnt[j] -= p;
p = 0;
break;
}
}
while (p)
{
for(int j = 0; j < 20; j++)
{
if(cnt[j])
{
int l = j * 5 + 6 - cnt[j];
while (p&&cnt[j])
{
printf("%d%c", l, p==1?'\n':' ');
l++;
cnt[j]--;
p--;
}
}
}
}
}
}
}