火车购票
目录
前言
最近我在做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()