CCF-CSP 火車購票

軟體開發 · 發表 2018-10-21 23:48:08

摘要： CCF-CSP 201609-2 火車購票題解問題描述試題編號：201609-2 試題名稱：火車購票時間限制：1.0s 記憶體限制：256.0MB 問題描述請實現一個鐵路購票系統的簡單座位分配演算法，來處理一節車廂的座位分配。假設一節車廂有20排、每一...

CCF-CSP 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。

題解1

題目的思路比較簡單，只是需要注意一些細節。我們初始一個20行5列的0陣列來代表車廂中的100個座位，用1和0來標識座位是否賣出。題目中要求，如果這幾張票可以安排在同一排編號相鄰的座位，則應該安排在編號最小的相鄰座位。否則應該安排在編號最小的幾個空座位中（不考慮是否相鄰）。因此，我們首先處理這種情況，每輸入一個購票數，我們從第1排開始安排查詢可以滿足該購票數的連續座位，並輸出座位號。如果到最後一排還沒有滿足的情況，那我們再從第1排開始安排零散的座位來滿足購票數，直到所有座位都被賣出。

n = int(input())# 購票指令數
lst = list(map(int, input().split()))# 購票數
c = [[0] * 5 for i in range(20)]# 標識是否已經賣出

for k in lst:# 遍歷購票數列表
for i in range(20):# 從第1排開始安排座位
if c[i].count(0) >= k:
count = k# 需要安排座位的票數
for j in range(5):# 遍歷第i行
if c[i][j] == 0 and count != 0:
c[i][j] = 1# 賣出一張，即i，j位置的
count -= 1# 需要安排座位的票數減1
print(i * 5 + j + 1, end=" ")# 輸出座位號
print()
break
elif i == 19 and c[i].count(0) < k:# 如果已經沒有可以分配的連續座位
count = k
for b in range(20):# 從第1排開始
if count == 0:
break
for j in range(5):# 把每一排的空座位安排出去
if c[b][j] == 0 and count != 0:
c[b][j] = 1
count -= 1
print(b * 5 + j + 1, end=" ")
if count == 0:
print()
break

題解2

由於一開始都是連續分配座位，所以每一排一開始都是連續分配的，那麼也就可以不使用二維陣列，只用一維陣列即可。初始化一個20個0元素的陣列，用於記錄每一排已經賣出火車票的張數。然後按照題解1的思路，先處理連續座位的情況，再處理非連續的情況。

n = int(input())# 購票指令數
lst = list(map(int, input().split()))# 購票數
c = [0] * 20# 標識每一排已經賣出的座位數

for k in lst:# 遍歷購票數列表
find = False
for i in range(20):# 從第1排開始安排座位，連續的情況
if 5 - c[i] - k >= 0:# 剩餘座位數可以滿足購票數
start = i * 5 + c[i] + 1
c[i] += k
for j in range(start, start + k):# 輸出座位號
print(j, end=" ")
print()
find = True# 標識為購票數已經得到滿足
break
if not find:# 非連續的情況
for i in range(20):
if k > 0 and c[i] < 5:# 非連續座位
start = i * 5 + c[i] + 1
length = 5 - c[i]# 剩餘座位數
if k > length:# 剩餘座位數無法滿足購票數
k -= length
c[i] = 5
for j in range(start, start + length):
print(j, end=" ")
else:# 可以滿足
c[i] += k
for j in range(start, start + k):
print(j, end=" ")
print()
break

知識點補充

列表初始化

除了基本的列表初始化方法外，我們還可以使用列表初始化表示式。列表初始化表示式為建立列表提供了一個簡單的方法。一個常見的應用就是對於一個序列，將其中的每個元素進行一些操作，產生新的列表，或者是從原序列按照一定條件創建出子列表。例如，使用map 函式與lambda 表示式產生平方數列表：

>>>list(map(lambda x: x**2, range(10)))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

或者使用如下等價形式：

>>>[x**2 for x in range(10)]
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

如題解1中使用列表初始化表示式來構建二維陣列：

c = [[0] * 5 for i in range(20)]

列表初始化表示式由方括號[] 包含，括號內以for 語句起始，後接任意個for 語句或if 語句。其結果是產生一個新的列表，列表內的元素為其中的for 語句或if 語句的執行結果。例如，以下表達式建立了一個列表，列表內的每個元素形如(x, y) ，其中x 和y 分別來自兩個列表，且x 與y 不相等。

>>>[(x, y) for x in [1, 2, 3] for y in [3, 1, 4] if x != y]
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

這種寫法等價於：

>>>combs = []
>>>for x in [1, 2, 3]:
...for y in [3, 1, 4]:
...if x != y:
...combs.append((x, y))
...
>>>combs
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

列表統計

List 的count() 方法用於統計某個元素在列表中出現的次數。語法格式如下：

list.count(obj)

引數列表：obj 列表中統計的物件。

返回值：返回元素在列表中出現的次數。

使用示例：

>>>lst = [1, 2, 3, 1]
>>>lst.count(1)
2

字串統計

str 也實現了count() 方法，它用於統計字串裡某個字元出現的次數。可選引數為在字串搜尋的開始與結束位置。語法格式如下：

str.count(sub, start= 0, end=len(string))

引數列表：

sub
start
end

返回值：該方法返回子字串在字串中出現的次數。