201609-2 火车购票 ccf

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    scanf("%d",&n);
    vector<int> v(20,5);
    for(int i=0;i<n;++i)
    {
        int x;
        scanf("%d",&x);
        for(int r=0;r<20&&x>0;++r)
        {
            if(v[r]>=x)//当前行空余座位大于需求x
            {
                int seat=r*5+5-v[r]+1;//开始空闲座位编号
                v[r]-=x;//x个订购，更新空余座位
                while(x--)
                {
                    cout<<seat++;
                    if(x>0) cout<<" ";
                    else cout<<endl;
                }
            }
        }
        if(x>0)//没有连续座位
        {
            for(int r=0;r<20&&x>0;++r)
            {
                while(v[r]>0&&x>0)
                {
                    int seat=r*5+5-v[r]+1;
                    cout<<seat++;
                    --v[r];
                    --x;
                    if(x>0)cout<<" ";
                    else cout<<endl;
                }
            }
        }
    }
    return 0;
}