CCF 201612-2 火车购票 JAVA实现

问题描述

　　请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
　　假设一节车厢有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;
/**
 * 大体思路是：每次购买车票时，求得每排座位剩余的座位数(20排，每排5个座位)，如果存在一排座位剩余座位数大于等于
 * 购买的车票数，则将这些购买的票分配到这一排，这样就符合分配在同一排编号相邻的座位。如果20排剩余座位数都比购买
 * 的票数少，则按顺序从第一排剩余座位开始分配购买的票数。
 * @author ooo小6
 *
 */
public class test2 {
	public static int[][] s;//存储每节车厢的座位
	public static int[] e;//存储车厢座位每排剩余的座位

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		s = new int[20][5];//0表示座位为空
		int n = scanner.nextInt();//购买指令数量
		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++) {//取出P数组中每次购买的票数进行分配座位
			for (int j = 0; j < 20; j++) {
				if (p[i] <= e[j]) {//找到一排空座位数大于等于购买的票数
					buy(p[i], j);//参数P[i]表示购买的票数，j表示这些票将分配在第j排(即第j排空座位数大于等于购买的票数)
					output();//输出本次购买的处理结果
					System.out.print("\n");
					e = has_empty(s);//每购买一次票后就得更新e中存储的每排空座位数
					break;
				}
				if (p[i] > e[j] && j == 19) {//20排座位中每排空座位数都比购买的票数少
					buy_left(p[i]);//将票按顺序分配到从第一排开始的空座位
					output();
					System.out.print("\n");
					e = has_empty(s);
					break;
				}
			}
		}
		scanner.close();
	}
	/*
	 * 分配座位
	 */
	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--;//票数减1
			}
			if (x == 0) {//分配完所有票数，退出
				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 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，否则每次购买完票调用该函数，会再次将之前购买好的座位号输出
				}
			}
		}
	}
}