【CSP201609-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。

解題思路：題目需要重點判斷的是乘客是否能買到連號的座位，以flag標記。用大小爲20的數組記錄每排已售出的座位，然後遍歷，看是否這一排有足夠的位置連號，有即輸出，flag設爲1，最後一項有格式要求。不能則再次遍歷單個輸出，i--是爲了防止漏掉這排的剩餘座位。

code：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
int a[20];
//int b[1001];
using namespace std;

int main()
{
    int n,p;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
        {
            int flag  = 0;
            scanf("%d",&p);

            for(int i=0;i<20;i++)
            {
                if((a[i]+p)<=5)
                {
                    for(int j =0;j<p;j++)
                    {
                        if(j==p-1)
                            printf("%d\n",i*5+j+a[i]+1);
                        else
                            printf("%d ",i*5+j+a[i]+1);
                    }
                    a[i]+=p;
                    flag = 1;
                    break;
                }
            }
            if(!flag)
            {
                int num = 0;
                for(int i= 0; i<20;i++)
                {
                    if(a[i]<5)
                    {
                        num++;
                        if(num==p){
                            printf("%d\n",a[i]+1+i*5);
                            break;
                        }
                        else
                            printf("%d ",a[i]+1+i*5);
                        a[i]++;
                        i--;

                    }
                }
            }
        }
    }
    return 0;
}