CCF CSP 201609-2 火车购票 C++实现 100分

问题描述
　　请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
　　假设一节车厢有20排、每一排5个座位。为方便起见，我们用1到100来给所有的座位编号，第一排是1到5号，第二排是6到10号，依次类推，第20排是96到100号。
　　购票时，一个人可能购一张或多张票，最多不超过5张。如果这几张票可以安排在同一排编号相邻的座位，则应该安排在编号最小的相邻座位。否则应该安排在编号最小的几个空座位中（不考虑是否相邻）。
　　假设初始时车票全部未被购买，现在给了一些购票指令，请你处理这些指令。
　　
输入格式
　　输入的第一行包含一个整数n，表示购票指令的数量。
　　第二行包含n个整数，每个整数p在1到5之间，表示要购入的票数，相邻的两个数之间使用一个空格分隔。
　　
输出格式
　　输出n行，每行对应一条指令的处理结果。
　　对于购票指令p，输出p张车票的编号，按从小到大排序。
　　
样例输入
4
2 5 4 2

样例输出
1 2
6 7 8 9 10
11 12 13 14
3 4

样例说明
　　1) 购2张票，得到座位1、2。
　　2) 购5张票，得到座位6至10。
　　3) 购4张票，得到座位11至14。
　　4) 购2张票，得到座位3、4。
　　
评测用例规模与约定
　　对于所有评测用例，1 ≤ n ≤ 100，所有购票数量之和不超过100。

思路

对于火车购票，我建立一个长度为21的数组，从b[1]——b[20] 代表火车的每排，值为5。
每次购票，需要先判断这个购票是否连坐，如果连坐的话，就从b[1]一直循环到b[20] 如果值大于购票数，则可以连坐，坐在一起就行，结束循环。
但是如果发现购票不能连坐，那么就从b[1]一直循环到b[20] ，找到b[i]>0的位置，那么就安排坐，同时继续找位置，一直到所有人都安排坐上位置。

#include 
#include
using namespace std;


int b[21];

int main()
{
	int n;
	cin >> n;
	int p[110];

	for (int i = 0; i < n; i++) {
		cin >> p[i];
	}


	for (int i = 0; i < 21; i++) {
		b[i]=5;
	}
	


	for (int i = 0; i < n; i++) {
		int flag = 0;
		for (int j = 1; j < 21; j++) {
					
			if (p[i] <= b[j]) {  // 可以连坐分配
				flag=1;
				for (int k = (j-1)* 5 + 1+5  - b[j]; k<= (j-1) * 5 + p[i]+ 5 - b[j]; k++) {
					cout << k << " ";
					
				}
				b[j] = b[j] - p[i];
				cout << endl;
				break;
			}
				
		}
		if (flag == 1)
			continue;

		int num = p[i];
		for (int j = 1; j < 21; j++) {   // 不可以连坐分配
			if (b[j] > 0) {
				int k;
				while (b[j] > 0 && num) {
					k = j*5 + 1 - b[j];
					cout << k << " ";
					num--;
					b[j]--;
				}
			}
			if(!num){
				break;
				cout << endl;
			}
		}
	}



	return 0;
}