CCF-火车购票

#include <iostream>
using namespace std;

int main()
{
	int n, num = 0;
	int arr1[101];//该数组表示需要购买的票数
	int arr2[20][5] = {};//二维数组表示整个车厢，座位为空时，数值为0
	cin >> n;
	for (int i = 0;i < n;i++)
	{
		cin >> arr1[i];
		for (int a = 0;a < 20;a++)
		{
			for (int b = 0;b < 5;b++)//num为每一排空座位的个数
			{
				if (arr2[a][b] == 0)
				{
					num++;
				}
			}
			if (arr1[i] <= num)//需要购买的票数小于一排中空座位的个数
			{
				for (int r = 0;r < arr1[i] + (5 - num);r++)
				{
					if (arr2[a][r] == 0)
					{
						arr2[a][r] = a * 5 + r + 1;
						cout << arr2[a][r] << \" \";
					}
				}
				cout << endl;
				num = 0;
				break;
			}
			else {
				num = 0;
				if (arr1[i] > num&&a == 19)//必须分开坐
				{
					for (int a = 0;a < 20;a++)
					{
						for (int b = 0;b < 5;b++)
						{
							if (arr2[a][b] == 0)//重新遍历
							{
								num++;
							}
						}
						if (arr1[i] < num)
						{
							for (int r = 0;r < arr1[i] + (5 - num);r++)
							{
								if (arr2[a][r] == 0)
								{
									arr2[a][r] = a * 5 + r + 1;
									cout << arr2[a][r] << \" \";
								}
							}
							cout << endl;
							break;
						}
						else
						{
							for (int r = 0;r < 5;r++)
							{
								if (arr2[a][r] == 0)
								{
									arr2[a][r] = a * 5 + r + 1;
									cout << arr2[a][r] << \" \";
								}
							}
						    arr1[i] = arr1[i] - num;
						    num = 0;
						}
					}
				}
			}
			num = 0;
		}
	}
}