CCF CSP 201609-2 火车购票 JAVA实现（90分）

 

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int a[] = new int[n];// 存放购票指令的数量
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        int seat[][] = new int[20][7];// 第6列存放起始空位，第七列存放空位数
        for (int i = 0; i < 20; i++) {
            for (int j = 0; j < 5; j++) {
                seat[i][6] = 5;
                if (seat[i][j] == 0) {
                    seat[i][5] = j;
                    break;
                }
            }
        }
        for (int i = 0; i < n; i++) {
            int col = 0;
            for (int j = 0; j < 20; j++) {
                if (a[i] <= seat[j][6]) {
                    col = j;
                    seat[j][6] = seat[j][6] - a[i];
                    for (int k = seat[j][5]; k < (seat[j][5] + a[i]); k++) {
                        seat[j][k] = 1;
                        System.out.print((j * 5) + (k + 1) + " ");
                    }
                    break;
                }

            }
            for (int s = 0; s < 5; s++) {
                if (seat[col][s] == 0) {
                    seat[col][5] = s;
                    break;
                }
            }

            System.out.println();

        }

    }

}