201609-2 火車購票

座位是否可坐的情況用一個二維數組布爾存儲，記爲seats，行列固定爲20*5。輸出座位編號的時候需要轉化下，如seats[i][j]的號碼應該是5*i + j + 1。座位找到後要及時break，提高時間效率

奉上java滿分代碼

import java.util.*;

public class Main{
    private static final int rows = 20, cols = 5;
    private static boolean[][] seats = new boolean[rows][cols];

    public static void main(String[] args){
       Scanner scanner = new Scanner(System.in);
       int n = Integer.parseInt(scanner.nextLine());
       int[] numbers = new int[n];
       String[] line = scanner.nextLine().split(" ");
       for(int i = 0; i < n; i++){
           numbers[i] = Integer.parseInt(line[i]);
       }
       scanner.close();

       for(int number : numbers){
           boolean isSeated = false;
           for(int i = 0; i < rows; i++){
               int j = 0;
               int count = 0;
               while (j < cols && seats[i][j])
                   j++;
               int k = j;
               while (k < cols && count < number){
                   count++;
                   k++;
               }
               if(count == number){
                   while (j < k){
                       printSeatNo(i, j++);
                   }
                   System.out.println();
                   isSeated = true;
                   break;
               }
           }
           if(!isSeated){
               int count = 0;
               for(int i = 0; i < rows && count < number; i++){
                   for(int j = 0; j < cols; j++){
                       if(!seats[i][j]){
                           count++;
                           printSeatNo(i, j);
                           if(count == number)
                               break;
                       }
                   }
               }
               System.out.println();
           }
       }
    }

    private static void printSeatNo(int i, int j){
        seats[i][j] = true;
        System.out.print((i * cols + j + 1) + " ");
    }
}