时间限制: 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。
import java.util.LinkedList; import java.util.Scanner; public class 火车购票 { public static void main(String[] args){ LinkedList[] queue = new LinkedList[20]; for(int i=0;i<queue.length;i++){ queue[i] = new LinkedList(); for(int j=0;j<5;j++) queue[i].offer(i*5+j+1); } Scanner input = new Scanner(System.in); int qty = input.nextInt(); for(int i=0;i<qty;i++){ int num = input.nextInt(); boolean flag = true; for(int j=0;j<queue.length;j++){ if(queue[j].size()>=num){ for(int k=0;k<num;k++) System.out.print (queue[j].poll()+" "); flag = false; break; } } if(flag){ int count = 0; for(int j=0;j<queue.length&&flag;j++){ while(!queue[j].isEmpty()){ System.out.print (queue[j].poll()+" "); count++; if(count==num){ flag = false; break; } } } } System.out.println(); } } }python 代码:
n = int(input()) matrix = [[e + 1 + 5 * i for e in range(5)] for i in range(20)] nums = [int(e) for e in input().split()] for num in nums: for line in matrix: if len(line) >= num: for i in range(num): print(line.pop(0), end=' ') break else: for line in matrix: while line and num > 0: print(line.pop(0), end=' ') num -= 1 if num == 0: break print()