CCF-火车购票（90）

#include<iostream>
#define len 100
using namespace std;
int main(void){
	int n;
	int count=0;
	int i,j,k,m;
	int a[len][2];
	int b[20][5];
	for(i=0;i<20;i++){
		for(j=0;j<5;j++){
			b[i][j]=0;
		}
	} 
	
	cin>>n;
	for(i=0;i<n;i++){
		cin>>a[i][0];
	}
	for(i=0;i<n;i++){
		a[i][1]=0;
	}
	
	for(k=0;k<n;k++){
		count=0;
		for(i=0;i<20;i++){
			count=0;
			
			for(j=0;j<5;j++){
				while(b[i][j]==0&&j<5){
					count++;
					j++;			
				} 
			}
			j--;
			
			if(count>=a[k][0]){
				for(m=j-count;a[k][0]>0;a[k][0]--){
					b[i][m]=1;
					cout<<i*5+m+1<<" ";
					m++;
				}
				a[k][1]=1;
				break;
			}
			
		}
		
		if(a[k][1]==0){
			for(i=0;i<20;i++){
				for(j=0;j<5;j++){
					if(b[i][j]==0){
						cout<<i*5+m+1<<" ";
						b[i][j]==1;
						a[k][0]--;
					}
					if(a[k][0]<=0){
						break;
					}		
				}
				if(a[k][0]<=0){
					break;
				}
			}
		}
		
		cout<<endl;
	}
	return 0;	
}
/*
4
2 5 4 2
*/