CCF 201609-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。

利用数组进行购票模拟

ps：当每一行所剩余的位置都不够p时，从前往后遍历空位置进行购买

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n, p, us[105], num[25];
int main()
{
while(scanf("%d", &n) != EOF)
{
4000
memset(us, 0, sizeof(us));
for(int i = 0; i <= 20; i ++) //每一行所剩余的票数
{
num[i] = 5;
}
int flag = 0;
while(n--)
{
scanf("%d", &p);
for(int i = 1; i <= 20; i++)
{
if(num[i] >= p) //每一行所余的票数足够
{
flag = 1;
num[i] -= p;
int cnt = 0;
for(int j = 1; j <= 5; j++) //计算得出当前可订的座位
{
if(!us[(i - 1) * 5 + j])
{
us[(i - 1) * 5 + j] = 1;
printf("%d", (i - 1) * 5 + j);
cnt++;
if(cnt != p)
printf(" ");
else
{
printf("\n");
break;
}
}
}
break;
}
}
if(flag)
{
flag = 0;
continue;
}
else //若每一行都没有足够的位置，则从第一行开始订剩余的不连贯的位置
{
int cnt = 0;
for(int i = 1; i <= 100; i++)
{
if(!us[i])
{
us[i] = 1;
printf("%d", i);
cnt++;
if(cnt == p)
{
printf("\n");
break;
}
else
printf(" ");
}
}
}
}
}
return 0;
}