CCF程式設計練習：火車購票（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.Arrays;
import java.util.Scanner;

public class Main {
	public static void main(String[] args){
		Scanner input=new Scanner(System.in);
		int n=input.nextInt();
		int[] by=new int[n];//購票資訊
		
		for(int i=0;i<n;i++)
			by[i]=input.nextInt();
		input.close();
		
		int[][] chair=new int[20][5];
		int[] count=new int[20];
		
		int m=1;
		for(int i=0;i<20;i++){
			for(int j=0;j<5;j++)
				chair[i][j]=m++;//初始化火車座位，按序編號
			count[i]=5;//初始化每排座位數
		}
		
		for(int i=0;i<n;i++){//遍歷購票資訊序列
			int j=0;
			while(j<20){
				if(by[i]<=count[j]){//先看看能不能買到同一排
					for(int h=0;h<5&&by[i]>0;h++)
						if(chair[j][h]!=0){
							System.out.print(chair[j][h]+" ");//輸出已買到的座位號
							
							chair[j][h]=0;//鎖定一個座位
							by[i]--;
							count[j]--;//當前排座位數量減一
						}
				}
				j++;
			}
			if(j>=20){//無法買到相鄰座位
				for(int t=0;t<20;t++)
					for(int r=0;r<5;r++)
						if(chair[t][r]!=0&&by[i]>0){
							System.out.print(chair[t][r]+" ");
							count[t]--;
							by[i]--;
							chair[t][r]=0;
						}
			}
			System.out.println();
		}
	}
}