[问题描述]
请实现一个铁路购票系统的简单座位分配算法,来处理一节车厢的座位分配。
假设一节车厢有20排、每一排5个座位。为方便起见,我们用1到100来给所有的座位编号,第一排是1到5号,第二排是6到10号,依次类推,第20排是96到100号。
购票时,一个人可能购一张或多张票,最多不超过5张。如果这几张票可以安排在同一排编号相邻的座位,则应该安排在编号最小的相邻座位。否则应该安排在编号最小的几个空座位中(不考虑是否相邻)。
假设初始时车票全部未被购买,现在给了一些购票指令,请你处理这些指令。
[输入格式]
对于所有评测用例,1 ≤ n ≤ 100,所有购票数量之和不超过100。
输入的第一行包含一个整数n,表示购票指令的数量。
第二行包含n个整数,每个整数p在1到5之间,表示要购入的票数,相邻的两个数之间使用一个空格分隔。
[输出格式]
输出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;
int a[22] = { 0 };//各排剩余座位数组
int b[105][6] = { 0 };//输出数组
bool f[105] = { 0 };//座位是否有人的标记数组
int main()
{
fill(a, a + 20, 5);//各排初始剩余座位数量记为5
cin >> n;
for (int i = 0; i < n; i++)
{
int cont;
cin >> cont;
bool flag = 1;
for (int j = 0; j < 20; j++)//连坐判断
{
if (a[j] >= cont)
{
for (int k = 0; k < cont; k++)
{
b[i][k] = (5 - a[j]) + k + 1 + j * 5;//填充输出数组
f[(5 - a[j]) + k + 1 + j * 5] = 1;//有人的座位标记
}
a[j] -= cont;//相应排的剩余座位减少
flag = 0;
break;
}
}
if (flag)//flag==1即不连坐
{
int k = 0;
for (int j = 1; j <= 100; j++)
{
if (!f[j])
{
b[i][k] = j;
k++;
f[j] = 1;
cont--;
a[j / 5] -= 1;
}
if (cont == 0)
break;
}
}
}
for (int i = 0; i < n; i++)
{
int j = 0;
while (b[i][j] != 0 && j < 5)
{
cout << b[i][j] << ' ';
j++;
}
cout << endl;
}
}
[运行结果]