铁轨---栈

原创于 2021-04-06 15:00:53 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #栈

记录一下，比较经典的关于栈的算法题

题目描述

某城市有一个火车站，铁轨铺设如图所示。有n节车厢从A方向驶入车站，按进站顺序编号1~n。你的任务是让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任何时刻，只有两种选择：A->C 和 C->B。

在这里插入图片描述
样例输入：
5
1 2 3 4 5
5
5 4 1 2 3
6
6 5 4 3 2 1

样例输出

YES
NO
YES

思路

在中转站C中，车厢符合后进先出的原则，称为LIFO表。其中LIFO代表Last In First Out。它还有一个更常用的名称--------栈。由于只有一端生长，实现只需要一个数组stack和栈顶指针。

代码

n=int(input())

numbers=list(map(int,input().split()))
A=1
B=0
top=0
stack=[0]*n
t=True
while   B<=n-1:
    if  A==numbers[B]:
        A+=1
        B+=1
    elif    stack[top]==numbers[B]:
        top-=1
        B+=1
    elif A<=n:
        top+=1
        stack[top]=A
        A+=1
    else:
        t=False
        break
if t:
    print("Yes")
else:
    print("No")