火车购票
一、问题描述
问题描述
请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
假设一节车厢有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。
# 火车购票
n = int(input())
a = list(map(int,input().split()))
result = []
d_status = {}
s_status = {}
for i in range(1,101):
s_status[i] = 0
for i in range(1,21):
d_status[i] = 5
for i in a:
temp = []
for j in d_status:
if i <= d_status[j]:
for m in range(5*j-4,5*j+1):
if s_status[m] == 0 and i> 0 :
s_status[m] = 1
d_status[j] -= 1
temp.append(m)
i -= 1
maxindex = max(d_status,key=d_status.get)
if i > d_status[maxindex]:
for j in s_status:
if s_status[j] == 0 and i >0:
s_status[j] = 1
d_status[(j-1)//5+1] -= 1
temp.append(j)
i -= 1
result.append(temp)
for i in result:
for j in i:
print(j,end=' ')
print('')
思路
- 用一个字典d_status来储存每一排的剩余作为情况，初始每一排都剩5；再用另一个字典s_status来储存所有位置的情况，0为空
- 输入数据之后，遍历购票指令，遍历每一排的情况，当这一排的剩余座位大于购票指令的数时，说明可以安排连续的座位，将s_status里的座位情况置1，然后排数的剩余座位减少
- 如果发现购票指令的数大于所有排数的剩余座位情况，遍历座位s_status，将0的位置安排座位，注意同时每一排的座位数会减少。