201609-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>
#include<vector>
#include<algorithm>
using namespace std;

int st[20][5];
int res[20][1];
int n;
vector<int> a; ///每个人买的票数


vector<vector<int> > seat( int i, vector<vector<int> > ans) {
	int j;
	bool flag = false;
	for (j = 0; j < 20; j++) {
		if (res[j][0] >= a[i]) {
			flag = true;
			int count = 0;
			for (int k = 0; k < 5; k++) {
				if (st[j][k] != 0 ) {
					ans[i].push_back(st[j][k]);
					st[j][k] = 0;
					res[j][0]--;
					++count;
				}
				if (count == a[i])
					break;
			}
		}
		if (flag == true)
			break;
	}
	if (flag==false) {
		int m = 0;
		for (int p = 0; p < 20; p++) {
			for (int q = 0; q < 5; q++) {
				if (st[p][q] != 0) {
					ans[i].push_back(st[p][q]);
					st[p][q] = 0;
					res[p][0]--;
					++m;
				}
			}
			if (m == a[i])
				break;
		}
	}
	return ans;
}
int main() {

	cin >> n;
	for (int i = 0; i < n; i++) {
		int v;
		cin >> v;
		a.push_back(v);
	}
	int m = 0;
	for (int i = 0; i < 20; i++) {  //1-100编号
		st[i][0] = ++m;
		st[i][1] = ++m;
		st[i][2] = ++m;
		st[i][3] = ++m;
		st[i][4] = ++m;
	}
	for (int i = 0; i < 20; i++) {
		res[i][0] = 5;
	}
	vector<vector<int> > ans(n);
	for(int i = 0; i < n; i++) {
		ans=seat(i,ans);
	}
	for (int i = 0; i < n; i++) {
		for (int t = 0; t < ans[i].size() - 1; t++) {
			cout << ans[i][t] << " ";
		}
		cout << ans[i][ans[i].size() - 1] << endl;
	}
	a.clear();
	ans.clear();
	return 0;
}

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

int main(){
	int n;  
	cin>>n;
	int a[20][5];//座位编号 
	int b[20]; //每排剩余座位 
	vector< vector<int> > ans(n);
	int number=1;
	for(int i=0;i<20;i++){
		for(int j=0;j<5;j++)
			a[i][j]=number++;
	}
	for(int i=0;i<20;i++)
		b[i]=5;
	
	int row=0; //第几排 
	for(int i=0;i<n;i++){
		int m;
		cin>>m; //每次买几张票
		int j;
		for(j=0;j<20;j++){
			if(b[j]>=m){
				row=j;
				for(int k=0;k<5;k++){
					if(a[row][k]!=0&&ans[i].size()<m){
						ans[i].push_back(a[row][k]);
						a[row][k]=0;
						--b[row];
					}
				}
				break;				
			}
		}
		if(j==20){
			for(int k=0;k<20;k++){
				for(int q=0;q<5;q++){
					if(a[k][q]!=0&&ans[i].size()<m){
						ans[i].push_back(a[k][q]);
					}
				}
			}
		} 
	}
	for(int i=0;i<ans.size();i++){
		for(int j=0;j<ans[i].size();j++)
			cout<<ans[i][j]<<" ";
		cout<<endl; 
	} 
/*	for(int i=0;i<20;i++){
		for(int j=0;j<5;j++)
			cout<<a[i][j]<<" ";
		cout<<endl; 
	}
*/
	return 0;
}