CCF CSP 201609-2 火车购票

问题描述

试题编号：	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。

解题的代码如下：

import java.util.Scanner;
/*
 * 火车购票
 */
public class Main {
	public static int[][] s;//存储每节车厢的座位
	public static int[] e;//存储车厢座位每排剩余的座位
	
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scanner = new Scanner(System.in);
		s = new int[20][5];//0表示座位为空
		int n = scanner.nextInt();//购买指令的数量,不超过5张
		int[] p = new int[n];//存储每次购买的票数
		for(int i=0;i<n;i++){
			p[i]=scanner.nextInt();
		}
		e = has_empty(s);//求得每排空座位的数量存到数组e中。数组e的下标和数组s的行坐标，即e[i]存的数为s中第i排剩余座位
		for(int i=0;i<n;i++){
			for(int j=0;j<20;j++){
				if(p[i]<=e[j]){	//找到一排空座位数大于等于购买的票数
					buy(p[i], j);
					output();//输出本次购买的结果
					System.out.println();
					e=has_empty(s);//每次购买一次票后就得更新e中存储的没排空座位数
					break;
				}
				if(p[i]>e[j]&&j==19){//20 排座位中每排空座位数都比购买的票数少
					buy_left(p[i]);//将票按顺序分配到从第一排开始的空座位
					output();
					System.out.println();
					e=has_empty(s);
					break;
				}
			}
		}
	}
	
	/*
	 * 20排座位中每排空座位数都比购买的票数少，调用该函数
	 */
	public static void buy_left(int sum){
		//按顺序搜索每个座位，是空座位就分配给购买的票
		for(int i=0;i<20;i++){
			for(int j=0;j<5;j++){
				if(s[i][j]==0&&sum>0){
					s[i][j]=1;
					sum--;
				}
			}
		}
	}
	
	
	/*
	 * 分配座位
	 */
	public static void buy(int x,int y){
		//将s中第y排座位中空座位分配
		for(int i=0;i<5;i++){
			if(s[y][i]==0){
				s[y][i]=1;//1表示该座位已分配
				x--;
			}
			if(x==0){//分配完所有票数，退出
				break;
			}
		}
	}
	
	
	/*
	 * 检索每一排座位数
	 */
	public static int[] has_empty(int[][] s){
		int e[] = new int[20];
		for(int i=0;i<20;i++){
			for(int j=0;j<5;j++){
				if(s[i][j]==0)
					e[i]++;
			}
		}
		return e;
	}
	
	
	/*
	 * 输出每次购买结果
	 */
	public static void output(){
		for(int i=0;i<20;i++){
			for(int j=0;j<5;j++){
				if(s[i][j]==1){
					System.out.print((5*i+j+1)+" ");//将i行j列座位换算成实际中第几个座位，并输出
					s[i][j]=2;//已经输出的座位号标记为2,否则每次购买完票调用该函数,会再次将之前购买好的座位号输出
				}
			}
		}
	}
	
}