python - 在 python 3 中查看堆栈

Question

问题:实现 peek(stack) 返回但不移除栈顶元素。如果列表为空，则返回 None。

试了很多次都没有成功，谁能帮帮忙？

我的尝试:

def peek_stack(stack):
    if stack == []:
        return None
    else:
        s= stack.copy()
        return s.pop(0)

Answer 1

如果你需要用你的方式解决这个问题，请使用return s.pop()而不是return s.pop(0)，因为 s.pop() 会弹出最后一个元素，但是 s.pop(0) 会弹出第一个元素...

顺便说一句，建议就这样实现(它可以避免复制你的堆栈，提高性能)

def peek_stack(stack):
    if stack:
        return stack[-1]    # this will get the last element of stack
    else:
        return None