问题描述
请实现一个铁路购票系统的简单座位分配算法,来处理一节车厢的座位分配。
假设一节车厢有20排、每一排5个座位。为方便起见,我们用1到100来给所有的座位编号,第一排是1到5号,第二排是6到10号,依次类推,第20排是96到100号。
购票时,一个人可能购一张或多张票,最多不超过5张。如果这几张票可以安排在同一排编号相邻的座位,则应该安排在编号最小的相邻座位。否则应该安排在编号最小的几个空座位中(不考虑是否相邻)。
假设初始时车票全部未被购买,现在给了一些购票指令,请你处理这些指令,输出购票结果。
例如:若一次购买2,5,4,2张票得到的购票结果为:
1) 购2张票,得到座位1、2。
2) 购5张票,得到座位6至10。
3) 购4张票,得到座位11至14。
4) 购2张票,得到座位3、4。
输入说明
输入由两行构成。
第一行包含一个整数n,表示购票指令的数量,1 ≤ n ≤ 100。
第二行包含n个整数,每个整数p在1到5之间,表示要购入的票数,相邻的两个整数之间使用一个空格分隔,所有购票数量之和不超过100。
输出说明
输出n行,每行对应一条购票指令的处理结果。即对于购票指令p,按从小到大排序输出p张车票的编号。
输入样例
4
2 5 4 2
输出样例
1 2
6 7 8 9 10
11 12 13 14
3 4
#include<iostream>
using namespace std;
int n,a[110],b[20][5];//数组开小会出现越界现象
int main(){
cin >> n;
for(int i = 0; i < n; i ++ ){
cin >> a[i];//每次购票的张数
}
for(int i = 0, t = 1; i < 20; i ++ ){
for(int j = 0; j < 5; j ++ ){
b[i][j] = t ++;
}
}
for(int q = 0; q < n; q ++ ){
if(a[q] == 1){
int temp = 0;
for(int i = 0, t = 1; i < 20; i ++ ){
for(int j = 0; j < 5; j ++ ){
if(b[i][j] != -1){//未安排座位
cout << b[i][j] << endl;//换行
b[i][j] = -1;//改变座位状态
temp = 1;//该次安排完成
break;
}
}
if(temp) break;//不存在失败情况
}
}else if(a[q] == 2){
int temp = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 4; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " " << b[i][j+1] << endl;
b[i][j] = -1, b[i][j+1] = -1;
temp = 1;
break;
}
}
if(temp) break;
}
if(temp != 1){
int p = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 5; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " ";
b[i][j] = -1;
p ++;
}
}
if(p == 2){
cout << endl;
break;
}
}
}
}else if(a[q] == 3){
int temp = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 3; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " " << b[i][j+1] << " " << b[i][j+2] << endl;
b[i][j] = -1, b[i][j+1] = -1, b[i][j+2] = -1;
temp = 1;
break;
}
}
if(temp) break;
}
if(temp != 1){
int p = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 5; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " ";
b[i][j] = -1;
p ++;
}
}
if(p == 3){
cout << endl;
break;
}
}
}
}else if(a[q] == 4){
int temp = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 2; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " " << b[i][j+1] << " " << b[i][j+2] << " " << b[i][j+3] <<endl;
b[i][j] = -1, b[i][j+1] = -1, b[i][j+2] = -1, b[i][j+3] = -1;
temp = 1;
break;
}
}
if(temp) break;
}
if(temp != 1){
int p = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 5; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " ";
b[i][j] = -1;
p ++;
}
}
if(p == 4){
cout << endl;
break;
}
}
}
}else if(a[q] == 5){
int temp = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 1; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " " << b[i][j+1] << " " << b[i][j+2] << " " << b[i][j+3] <<" " << b[i][j+4] <<endl;
b[i][j] = -1, b[i][j+1] = -1, b[i][j+2] = -1, b[i][j+3] = -1, b[i][j+4] = -1;
temp = 1;
break;
}
}
if(temp) break;
}
if(temp != 1){
int p = 0;
for(int i = 0; i < 20; i ++){
for(int j = 0; j < 5; j ++){
if(b[i][j] != -1){
cout << b[i][j] << " ";
b[i][j] = -1;
p ++;
}
}
if(p == 5){
cout << endl;
break;
}
}
}
}
}
return 0;
}