Python入门习题（36）——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. 从左到右依序处理各次购票。对每一次购票，先满足同一排相邻的要求，如果无法满足，就安排编号最小的空座位。
2. 如何知道能否满足同一排相邻要求？答案是，用seats_remain列表（即数组）记下每一排的空余座位数，从第一排到第20排依次比较，如果购票数小于这一排剩余座位数，那么满足。全部比较后，各排剩余座位数都小于购票数，意味着不满足。
3. 满足同一排相邻要求的话，假定是第k排符合要求，根据第k排剩余座位数r可以得出这一次购票分配到的座位编号。这些座位编号依次是：(k - 1) * 5 + 5 - r + 1, (k - 1) * 5 + 5 - r + 2, …
4. 不满足同一排相邻要求的话，就分别为每一张票找最小编号的空座位。
5. 无需记住各排已经分配的座位号。因为根据任一排的剩余座位数，能够推导出这一排已经分配的座位号和剩余的座位号。已分配的座位号总是这一排中最小的。

参考答案

def form_seat_number(row, remain, t):
    '''第row行有remain个空座位，分配t个座位，得到这些座位的编号。'''
    start = row * 5 + 5 - remain + 1
    seat_numbers = [start + i for i in range(t)]
    return seat_numbers

def find_seats(ticket, seats_remain):
    '''找ticket个座位，先满足同一排相邻。seats_remain记录了各排剩余空位数量'''
    for row, remain in enumerate(seats_remain):
        if ticket <= remain:
            seats_remain[row] -= ticket
            return form_seat_number(row, remain, ticket)  #满足同一排相邻的要求

    '''没找到同一排相邻座位，分配编号小的空座位'''
    seat_numbers = []  #记录分配的空座位
    row = 0
    for i in range(ticket):
        while seats_remain[row] == 0:
            row += 1
        seat_numbers.append(row * 5 + 5 - seats_remain[row] + 1)
        seats_remain[row] -= 1

    return seat_numbers

seats_remain = [5] * 20

num_ticket = int(input())
tickets = [int(s) for s in input().split()]
seat_arrangment = []
for t in tickets:
    t_arrangement = find_seats(t, seats_remain)
    seat_arrangment.append(t_arrangement)

for sa in seat_arrangment:
    for a in sa:
        print(a, end = ' ')
    print()

测试用例

1. 题目描述给出的测试用例覆盖了满足同一排相邻要求的情形。第一排位两次购票提供了座位。

2. 无法满足同一排相邻要求的情形。
   样例输入
   21
   4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 5
   样例输出
   1 2 3 4
   6 7 8 9
   11 12 13 14
   16 17 18 19
   21 22 23 24
   26 27 28 29
   31 32 33 34
   36 37 38 39
   41 42 43 44
   46 47 48 49
   51 52 53 54
   56 57 58 59
   61 62 63 64
   66 67 68 69
   71 72 73 74
   76 77 78 79
   81 82 83 84
   86 87 88 89
   91 92 93 94
   96 97 98 99
   5 10 15 20 25
   只要想到了，这一个测试用例是容易构造的。输出结果的各行是有规律的，比对难度小。对于第21次购票，所购买的5张票分配在第1, 2, 3, 4, 5行。每一排提供一张。

3. 无法满足同一排相邻要求的情形。某一排提供了多张最小编号的座位。下面这一组测试用例是在第2组测试用例的基础上稍加改动得到的。
   样例输入
   21
   3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 5
   样例输出
   1 2 3
   6 7 8 9
   11 12 13 14
   16 17 18 19
   21 22 23 24
   26 27 28 29
   31 32 33 34
   36 37 38 39
   41 42 43 44
   46 47 48 49
   51 52 53 54
   56 57 58 59
   61 62 63 64
   66 67 68 69
   71 72 73 74
   76 77 78 79
   81 82 83 84
   86 87 88 89
   91 92 93 94
   96 97 98 99
   4 5 10 15 20

小结

1. 不考虑编程实现的话，你会如何解答本题？你构造的解法估计与“解题思路”一节给出的解法相差不大。编程实现要做的是，构造可以映射为程序代码的算法和设计若干数据结构记住算法需要的信息。
2. 多运用函数编写代码是好习惯。上面给出的代码中，抽取出find_seats()函数是有益的。这一函数把“从左到右依序处理各次购票”和“为一次购票分配座位”这两项任务分离开来。不分离开来的话，将有两重循环的纠缠，会更费脑力，且增加出错几率。