csp认证 火车购票

#include<iostream>
#include<cmath>
using namespace std;
int n;
int a[21][7] = { 0 };				//记录每个座位是否已经售出，并记录每一排还剩几个座位
int main()
{
	cin >> n;
	for (int i = 1; i <= 20; i++)	//每一排初始为5个空座位
	{
		a[i][6] = 5;
	}
	while (n--)
	{
		int temp;
		cin >> temp;
		for (int i = 1; i <= 20; i++)//遍历每一个座位
		{
			int flag = 0;
			int flag1 = 0;
			if (temp <= a[i][6])	//可以连坐
			{
				a[i][6] -= temp;	//减去售出的座位
				for (int j = 1; j <= 5; j++)
				{
					//cout << "ok";
					if (a[i][j] == 0)
					{
						a[i][j] = 1;
						temp--;
						cout << (i-1)*5+j << " ";
						//cout << "ok:" << i << endl;
						flag = 1;
						if (temp == 0)
						{
							break;
						}
					}
				}
			}
			if (flag == 1)
			{
				cout << endl;
				break;
			}
			if (i == 20)	//如果遍历到了最后一排还不能连坐
			{
				for (int k = 1; k <= 20; k++)
				{
					for (int x = 1; x <= 5; x++)
					{
						if (a[k][x] == 0)
						{
							cout << (k-1) * 5 + x << " ";
							temp--;
						}
					}
					if (temp == 0)
					{
						flag1 = 1;
						cout << endl;
						break;
					}
				}
			}
		}
	}
	return 0;
}