火车购票

一、问题描述

问题描述
　　　请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
　　假设一节车厢有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('')

思路

1. 用一个字典d_status来储存每一排的剩余作为情况，初始每一排都剩5；再用另一个字典s_status来储存所有位置的情况，0为空
2. 输入数据之后，遍历购票指令，遍历每一排的情况，当这一排的剩余座位大于购票指令的数时，说明可以安排连续的座位，将s_status里的座位情况置1，然后排数的剩余座位减少
3. 如果发现购票指令的数大于所有排数的剩余座位情况，遍历座位s_status，将0的位置安排座位，注意同时每一排的座位数会减少。

[Python]CCF——火车购票(201609-2)

原文：https://www.cnblogs.com/SavvyM/p/11785385.html