算法竞赛入门经典 例题6-2 铁轨（C、python）

同个人网站 https://www.serendipper-x.cn/，欢迎访问 ！

问题描述：
某城市有一个火车站，铁轨铺设如图所示。有n节车厢从A方向驶入车站，按进站顺序编号为 1~n 。你的任务是判断是否能让它们按照某种特定的顺序进入 B 方向的铁轨并驶出车站。例如，出栈顺序（5 4 1 2 3）是不可能的，但（5 4 3 2 1）是可能的。

为了重组车厢，你可以借助中转站 C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入 C 的车厢必须按照相反的顺序驶出 C。对于每个车厢，一旦从 A 移入 C ，就不能再回到 A 了；一旦从 C 移入 B，就不能回到 C 了。换句话说，在任意时刻，只能由两种选择： A -> C 和 C -> B。

分析： 在这里中转站 C 符合先进后出的原则，可以把它当作是栈。

C:

#include<cstdio>
#include<stack>
using namespace std;
const int MAXN = 1000 + 10;

int n, target[MAXN];

int main(){
	while(scanf("%d", &n) == 1){
		stack<int> s;
		int A = 1, B = 1;
		for(int i = 1; i <= n; i++)
			scanf("%d", &target[i]);
		int ok = 1;
		while(B <= n){
			if(A == target[B]){
				A++;B++;
			}
			else if(!s.empty() && s.top() == target[B]){
				s.pop();B++;
			}
			else if(A <= n)
				s.push(A++);
			else{
				ok = 0;
				break;
			}
		}
		printf("%s\n", ok ? "Yes" : "No");
	}
	return 0;
}

Python:

n = int(input())
target = list(map(int, input().split()))  # 测试队列
stack = []

ok, A, B = 1, 1, 0
while B < n:
    if A == target[B]:
        A += 1
        B += 1
    elif len(stack) != 0 and stack[len(stack)-1] == target[B]:
        stack.pop()
        B += 1
    elif A <= n:
        stack.append(A)
        A += 1
    else:
        ok = 0
        break
if ok == 1:
    print ("Yes")
else:
    print ("No")