问题描述
试题编号: | 201609-2 |
试题名称: | 火车购票 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 请实现一个铁路购票系统的简单座位分配算法,来处理一节车厢的座位分配。 输入格式 输入的第一行包含一个整数n,表示购票指令的数量。 输出格式 输出n行,每行对应一条指令的处理结果。 样例输入 4 样例输出 1 2 样例说明 1) 购2张票,得到座位1、2。 评测用例规模与约定 对于所有评测用例,1 ≤ n ≤ 100,所有购票数量之和不超过100。 |
使用一个全局变量存储当前车厢内的空座位情况,0表示为空,1表示已经被售出
依次根据需求来计算分配的座位
#include<iostream>
#include<vector>
#include<string>
#include<bitset>
#include<map>
#include<algorithm>
#include<queue>
using namespace std;
int train[20][5] = { 0 };
int avaliable(int t[])//判断该排还有几个空座位
{
int tem = 0;
for (int i = 0; i <5; i++)
{
if (t[i] == 0)
tem++;
}
return tem;
}
vector<int> get_ticket(int n)//需要n个座位的情况下,返回值为所有的座位编号
{
vector<int> rt;
for (int i = 0; i < 20; i++)//从第一排开始遍历
{
if (avaliable(train[i]) >= n) //如果当前排拥有足够的座位
{
for (int j = 0; j < 5; j++)
{
if (train[i][j] == 0) //遇到空座位就分配
{
train[i][j] = 1;
rt.push_back(5 * i + j + 1);
n--;
if(n==0)
return rt;//分配完n个座位后返回
}
}
}
else
continue;
}
for (int i = 0; i < 20; i++)//如果遍历了所有的排都没有连续的n个座位
{
for (int j = 0; j < 5; j++) //从前往后找到n个空的座位
{
if (train[i][j] == 0)
{
train[i][j] = 1;//标记该座位已经被售出
rt.push_back(5 * i + j + 1);
n--;
if (n == 0)
return rt;//返回结果
}
}
}
}
int main()
{
int n;
cin >> n;
vector<int> need(n,0);
for (int i = 0; i < n; i++)
{
cin >> need[i];
}
for (int i = 0; i < need.size(); i++)//依次处理购票需求
{
vector<int> result = get_ticket(need[i]);
for (int j = 0; j < result.size(); j++)
{
cout << result[j]<<" ";
}
cout << endl;
}
}