CCF201612-2火车购票

问题描述

　　请实现一个铁路购票系统的简单座位分配算法，来处理一节车厢的座位分配。
　　假设一节车厢有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。

解题思路

可以先设定每排座位的初始座位量，然后遍历每排，当每排座位数大于或等于购票数时即售出该排的座位，售完后，改排座位减去购票数等于剩余的座位数，每次遍历从小到大即可

#include<iostream>
using namespace std;

int main()
{
	int line[21] = {0,5,5,5,5, 5,5,5,5,5, 5,5,5,5,5, 5,5,5,5,5,5, };//初始化每一排座位的数量
	int seat[101] = {0};//初始化每个座位的状态，0为没售出，1为已售出
	int n, p, t = 0,m=0;//t为每次售票的座位号，m为每次售票数
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> p;
		if (p == 5) {
			m = 0;
			for (int j = 1; j < 21; j++) //循环判断
				if (line[j] == 5) {
					for (int k = 1; k <= 5; k++) {
						t = (j - 1) * 5 + k;
						if (m != 4) { cout << t << ' '; m++; }
						else  cout << t << endl;
						seat[t] = 1;
					}
					line[j] = 0; break;//售完后设定每排剩余座位
				}
		}
		else if (p == 4) {
			m = 0;
			for (int j = 1; j < 21; j++) 
				if (line[j] >= 4) {
					for (int k = 1; k <= 4; k++) {
						t = (j - 1) * 5 + k;
						while (seat[t] != 0)
							t++;
						if (m!=3) { cout << t << ' '; m++; }
						else cout << t<<endl;
						seat[t] = 1;
					}
					line[j] -= 4; break;
				}				
		}
		else if (p == 3) {
			m = 0;
			for (int j = 1; j < 21; j++) 
				if (line[j] >= 3) {
					for (int k = 1; k <= 3; k++) {
						t = (j - 1) * 5 + k;
						while (seat[t] != 0)
							t++;
						if (m != 2) { cout << t << ' '; m++; }
						else cout << t << endl;
						seat[t] = 1;
					}
					line[j] -= 3; break;
				}
		}
		else if (p == 2) {
			m = 0;
			for (int j = 1; j < 21; j++) 
				if (line[j] >= 2) {
					for (int k = 1; k <= 2; k++) {
						t = (j - 1) * 5 + k;
						while (seat[t] != 0)
							t++;
						if (m !=1) { cout << t << ' '; m++; }
						else cout << t << endl;
						seat[t] = 1;
					}
					line[j] -= 2; break;
				}
		}
		else if (p == 1) {
			m = 0;
			for (int j = 1; j < 21; j++) 
				if (line[j] >= 1) {
					for (int k = 1; k <= 1; k++) {
						t = (j - 1) * 5 + k;
						while (seat[t] != 0)
							t++;
						cout << t << endl;
						seat[t] = 1;
					}
					line[j] -= 1; break;
				}
		}	
	}
	return 0;
}