CCF——火车购票

#CCF——火车购票

定义了两个函数，一个是desk（），当座位不大于5时取得符合要求的座位号 （第一个）；另一个out（）是当没有满足连续相邻座位时直接选择最小的座位号。

个人觉得我的代码都没咋用到复杂的东西，算法还是比较容易想到的啦~

#include<iostream>
#include<queue>
using namespace std;

int a[20][5],n;

int desk(int num){
 for(int i=0;i<20;i++){
  for(int j=0;j<=5-num;j++){
   int flag=0;
   for(int k=0;k<num;k++){
    if(a[i][j+k]==1) flag++;
   }
   if(flag==num){
    return (5*i+j+1);
   }
  }
 }
 return -1;
}

int out(int num){
 for(int i=0;i<20;i++){
  for(int j=0;j<5;j++){
   if(num){
    if(a[i][j]==1){
     cout<<(5*i+j+1)<<" ";
     num--;
     a[i][j]=0;
    }
   }
  }
 }
 cout<<endl;
 return 0;
}

int main(){
 int num,i,j;
 queue<int> q;
 cin>>n;
 for(i=0;i<n;i++){
  cin>>num;
  q.push(num); 
 }
 for(i=0;i<20;i++){
  for(j=0;j<5;j++) a[i][j]=1;
 }
 while(!q.empty()){
  int first;
  num=q.front();
  q.pop();
  if(num<=5){
   first=desk(num);
   if(first==-1) out(num);
   else{
    for(j=0;j<num;j++){
     int x,y;
     x=(first+j)/5;
     y=(first+j)%5-1;
     cout<<first+j<<" ";
     a[x][y]=0;
    }
    cout<<endl;
   }
  }
  else{
   out(num);
  }
 }
 return 0;
}