CCF — 火车购票（python实现：100分）

2016.09.2 火车购票

目录

前言

一、问题描述

二、问题分析

三、程序说明

---

前言

最近我在做CCF的题目，也打算把每道题的解题思路都写到博客上来，希望能帮助到也在做CCF题目的读者们，希望你们通过本文能有所提示，帮助大家提升编程能力。另外有个很好的想法就是，建议读者们可以先看一下问题分析这一部分，然后自己再整理一下思路，重新做一遍，最后再参考代码，我想这样会更有收获。

 

一、问题描述

问题描述

请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。

假设一节车厢有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。

二、问题分析

这道题的解题逻辑不是很难，但编程时有坑，现在先讲一下大致的解题思路：

①先用初始化数据：20行5列的num座位列表、一个存储当前每一行的座位数的seat一维列表（每一行初始值为5）、存储每一条购票指令的购票数的m一维列表、一个存储每一条购票指令所选择的座位编号的res二维列表

②遍历m列表，按顺序执行购票指令；先从seat列表中找到哪一行的座位数满足购票数的，将行数赋值给一个变量p，然后跳出seat循环

③这里补充一下：p初始为-1；当找到对应的行数时赋值为对应的行数（即行下标），如果找不到说明没能找到相邻的座位的，这时候需要从头插入座位（即不考虑相邻情况）

④判断p值，如果p != -1，则遍历该行的列数，把不等于0的座位编号存储到res列表中，同时把num的该座位编号置为0，表示已经被订购了，直到满足购票数的值，即可退出循环

⑤如果p == -1，说明每一行的座位都已不能安排相邻的座位，此时需要从头开始遍历num列表，找出空位（判断num[i][j] != 0）就插入，直到满足购票数的值，即可退出循环

⑥最后输出res列表的元素即可。

 

三、程序说明

n：购票指令的数量

m：存储每条购票指令的购票数量的列表

num：初始化一节车厢的座位列表（20行5列）

seat：长度为20的一维数组，初始值为5，存储当前每一行剩余的座位数

p：选座位的时候用到的变量，记录当前哪一行的座位数是够用的

res：是一个n行的二维列表，存储每一条购票指令所选择的座位编号

a：是res列表的行下标

python程序如下（100分）：

# 火车购票

n = int(input())
m = [0]*n
s = input().split()
for i in range(n):
    m[i] = int(s[i])

num = []  # 座位列表
for i in range(20):
    temp = []
    for j in range(5):
        temp.append(i*5+j+1)
    num.append(temp)

seat = [5 for i in range(20)]  # 每一行剩余的座位数
p = -1  # 记录哪一行的座位数够用

res = [[] for i in range(n)]  # 存储购票位置
a = 0  # res的下标

for i in m:
    # 找出p,找到即退出循环
    for j in range(20):
        if seat[j] >= i:
            p = j
            break
    # 选座位
    if p != -1:
        for k in range(5):
            if i == 0:
                break
            if num[p][k] != 0:
                res[a].append( num[p][k] )
                num[p][k] = 0
                seat[p] -= 1
                i -= 1
        a += 1
    else:
        # 找不到行数，即要插空位了
        for j in range(20):
            for k in range(5):
                if i == 0:
                    a += 1 # 此处与上面部分不同，是要订购完车票才+1
                    break
                if num[j][k] != 0:
                    res[a].append( num[j][k] )
                    num[j][k] = 0
                    seat[j] -= 1
                    i -= 1
            if i == 0:
                break
    p = -1 # 每一次循环后都要回复p的初始值

for i in res:
    for j in i:
        print(j,end=' ')
    print()