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。
題解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
返回值:該方法返回子字串在字串中出現的次數。
使用示例:
>>>s = "I Love You" >>>s.count('o')# 統計字元o出現的次數 2 >>>s.count('o', 4)# 從索引為4的字元開始統計 1