可以利用二维向量来模拟此过程,每一个内部向量按照先后次序将5-1,10-6,15-11,……,100-96压入(栈操作)。然后进行模拟。如果当前排剩余空座位(栈内元素个数)满足购票数则进行相应次数的出栈操作,否则按次序遍历各个栈然后输出。
具体代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
using namespace std;
int main()
{
vector<vector<int>> rows; //存放未被分配的座位号
for(int i = 0; i < 20; i++) //初始化
{
vector<int> temp(5);
iota(temp.begin(), temp.end(), 1 + 5 * i);
reverse(temp.begin(), temp.end());
rows.push_back(temp);
}
int n;
cin >> n;
while(n--)
{
unsigned k;
cin >> k;
for(auto &r : rows) //逐排查找,注意这里必须要用引用!
{
if(k <= r.size()) //当前排剩余空座不小于所买票数则依次输出座位号
{
for(; k > 0; k--)
{
cout << r.back() << ' ';
r.pop_back();
}
break;
}
}
for(int i = 0; k > 0; i++) //没有满足条件的连号座位则按序输出
{
while(!rows[i].empty())
{
cout << rows[i].back() << ' ';
rows[i].pop_back();
k--;
}
}
cout << endl;
}
return 0;
}