CCF 20160902 火车购票 Java

import java.util.*;
public class T20160902 {
	public static void main(String[] args) {
		Scanner in=new Scanner(System.in);
		int n=in.nextInt();
		int p[]=new int[n];
		for(int i=0;i<n;i++){
			p[i]=in.nextInt();
		}
		int zuo[]=new int[100];
		int kong[]=new int[20];
		for(int i=0;i<20;i++){
			kong[i]=5;
		}
		for(int i=0;i<100;i++){
			zuo[i]=i+1;
		}
		for(int i=0;i<n;i++){
			for(int j=0;j<20;j++){
				if(p[i]<=kong[j]){
					int l=j*5-kong[j]+5;
					int r=l+p[i];
					for(int k=l;k<r;k++){
						System.out.print(zuo[k]+" ");
					}
					System.out.println();
					kong[j]=kong[j]-p[i];
					break;
				}
				if(kong[19]<5&&p[i]>kong[j]){
					int l=j*5-kong[j]+5;
					int r=(j+1)*5;
					for(int k=l;k<r;k++){
						System.out.print(zuo[k]+" ");
					}
					p[i]=p[i]-kong[j];
				}
				
			}
		}
	}
}