CCF-CSP 2016 9 2-火车购票

麻了，有点费脑，这题尤其要注意if语句的条件这些

#include<iostream>
using namespace std;
int main(){
	int n,m;
	int p[20][6]={0};//座位没人为0，有人坐为1 
	for(int i=0;i<20;i++){
		p[i][0]=5;//剩余几个连续空座位 
	}
	cin>>n;
	for(int i=0;i<n;i++){
		int flag=0;//是否能连坐 
		cin>>m;
		for(int j=0;j<20;j++){
			if(p[j][0]>=m){
				p[j][0]-=m;
				for(int k=1;k<6;k++){
					if(m!=0){
						if(p[j][k]==0){
							p[j][k]=1;
							m--;
							cout<<5*j+k<<" ";
						}
					}
					else{
						break;
					}
				}
				cout<<endl;
				flag=1;
				break;
			}
		}
		if(flag==0){
			for(int j=0;j<20;j++){
				for(int k=1;k<6;k++){
					if(m!=0){
						if(p[j][k]==0){
							p[j][k]=1;
							p[j][0]--;
							m--;
							cout<<5*j+k<<" ";
						}
					}
					else{
						break;
					}
				}
			}
			cout<<endl;
		}
	}
}