CSP 201609-2 火车购票

一、问题链接

CSP201609-2 火车购票

二、参考代码

#include <bits/stdc++.h>
using namespace std;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n,p;
	cin>>n;
	vector<vector<int>> a(20);
	for(int i=0; i<n; i++) {
		cin>>p;
		for(int j=0; j<20; j++) {
			if(5 - a[j].size() >= p) { //有连坐
				while(p--) { //处理
					cout<<j*5+a[j].size()+1<<" ";
					a[j].push_back(1);
				}
				goto loop;
			}
		}
		for(int j=0; j<20; j++) {
			while(a[j].size()<5) {//有空座
				cout<<j*5+a[j].size()+1<<" ";
				a[j].push_back(1);
				p--;
				if(p==0) goto loop;
			}
		}
		loop:;
		cout<<"\n";
	}
	return 0;
}