题意
给定火车按照编号1~n的顺序进入中转站,判断某种特定的出站情况是不是正确的。
思路
中转站的结构就是一个栈,先进后出,用一个栈去模拟就好。对于给定的情况A从头到尾的遍历,对当前的数a进行分析,假设前面的数都是合理的,可以得到的,那么a只有两种可能,一种是在栈顶,一种是还没入栈。这样,我们需要一个标记来表示下一个还没入栈的数b。于是遍历时,对于每一个数字都判断一下是不是和栈顶相等就好。若相等,则栈顶出栈,不相等,把b及其后面小于a的所有数字按顺序入栈,再将b设为a+1。如果发现栈顶也不相等,且b已经大于了a,说明a在栈的内部压着,这时候我们判定为该情况无法实现。
代码
#include<cstdio>
#include<stack>
using namespace std;
int r[1005];
int main()
{
stack<int>rails;
int n;
scanf("%d",&n);
while(n)
{
scanf("%d",&r[0]);
while(r[0])
{
int con=0;
while(con<r[0]-1)rails.push(++con);
++con;
int flag=0;
for(int i=1;i<n;++i)
{
scanf("%d",&r[i]);
while(con<r[i])rails.push(++con);
if(con>=r[i])
{
if(rails.top()==r[i])rails.pop();
else flag=1;
}
}
if(flag)printf("No\n");
else printf("Yes\n");
scanf("%d",&r[0]);
}
printf("\n");
scanf("%d",&n);
}
return 0;
}