← 返回题库
初级

Pop Sequence

未完成
初级参考 完整示例代码供参考,建议自己理解后重新输入
def solve(m, sequence):
    m = int(m)
    sequence = list(map(int, sequence.split(',')))
    stack = []
    current = 1
    for num in sequence:
        while current <= num and len(stack) < m:
            stack.append(current)
            current += 1
        if not stack or stack[-1] != num:
            print('NO')
            return
        stack.pop()
    print('YES')

示例

输入
solve(5, '1,2,3,4,5')
期望输出
YES
Python 代码 🔒 登录后使用
🔒

登录后即可练习

注册免费账号,在浏览器中直接运行 Python 代码