CCF认证2016092-火车购票

本人初学，水平有限，若有不足，恳请赐教！

可以利用二维向量来模拟此过程，每一个内部向量按照先后次序将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;
}