CCF 201609-2 火车购票 100分

#include<bits/stdc++.h>
using namespace std;

int main() {
	int pos[20][5],tmp=1;
	for(int i=0; i<20; i++) //初始化座位数组
		for(int j=0; j<5; j++)
			pos[i][j]=tmp++;
	int n,count[101];
	cin>>n;
	for(int i=0; i<n; i++) //记录购入的票数
		cin>>count[i];
		
	int surplus; //记录第一轮循环后剩余的次数
	for(int k=0; k<n; k++) { //第一轮循环将连座票卖完
		for(int i=0; i<20; i++) {
			int need=0; //需要几连座的票
			for(int j=0; j<5; j++) {
				if(pos[i][j]!=-1)
					need++;
				if(need==count[k]) {
					for(int l=j-need+1; l<j+1; l++) {
						cout<<pos[i][l]<<" ";
						pos[i][l]=-1; //卖过的座位设为-1
					}
					cout<<endl;
					i=21;
					surplus=k; 
					break;
				}
			}
		}
	}

	for(int k=surplus+1; k<n; k++) { //第二轮循环将剩余零散票卖完
		int need=0;
		for(int i=0; i<20; i++) {
			for(int j=0; j<5; j++) {
				if(pos[i][j]!=-1) {
					need++;
					cout<<pos[i][j]<<" ";
					pos[i][j]=-1;
				}
				if(need==count[k]) {
					cout<<endl;
					i=21;
					break;
				}
			}
		}
	}
	return 0;
}