CCF CSP 201609-2 火車購票（Java-100分）

試題編號：	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。

問題描述：實現一個鐵路購票系統的簡單座位分配算法，來處理一節車廂的座位分配。

問題分析：該問題有個陷阱，就是容易忘記考慮當同排的連號票均已售出時，非同排票的售出處理。忽略這個場景，程序只能得90分。程序採用數組表示每排已售出的票數，這種思路方便給出兩種情況下的判斷條件。

提交後得100分的Java語言程序如下：（提交時去除所有註釋，否則容易導致編譯錯誤）

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		int M = 20;
		int a[] = new int[M];

		int n, p;
		Scanner sc = new Scanner(System.in);
		n = Integer.valueOf(sc.nextLine());

		while (n-- != 0) {
			p = sc.nextInt();
			boolean flag = false;
			// 首先判斷是否能夠買到同一排相鄰的座位
			for (int i = 0; i < 20; i++) {
				if (a[i] + p <= 5) {// a[i]表示第i組的5個位置已售出票數
					for (int j = 0; j < p; j++) {
						if (j == p - 1)
							System.out.print(i * 5 + a[i] + j + 1);
						else
							System.out.print(i * 5 + a[i] + j + 1 + " ");
					}
					a[i] += p;
					flag = true;
					break;
				}
			}
			// 如果不能買到同一排相鄰的座位，就依次尋找空的座位
			if (!flag) {
				int num = 0;
				for (int i = 0; i < 20; i++) {
					if (a[i] < 5) {
						num++;
						if (num == p) {
							System.out.print(i * 5 + a[i] + 1);
							break;
						} else
							System.out.print(i * 5 + a[i] + 1 + " ");
						a[i]++;
						i--;// 因爲在這一排還可能有空座位剩餘，故回退，當a[i]=5則不用回退
					}
				}
			}
			System.out.println();
		}

		sc.close();
	}
}