车厢调度(python)

解题思路:

注意事项:

参考代码:

def trainSchedule(n, order):

    stack = []

    next_car = 1

    for car in order:

        while (len(stack) == 0 or stack[-1] != car) and next_car <= n:

            stack.append(next_car)

            next_car += 1

        if stack[-1] == car:

            stack.pop()

        else:

            return "NO"

    return "YES"

# 读取输入

n = int(input())

order = list(map(int, input().split()))

# 调度车厢并判断能否得到指定的顺序

result = trainSchedule(n, order)

# 输出结果

print(result)