如何让用户输入'XXX'结束？

Question

stack.append('a')
stack.append('b')
stack.append('c')
stack.append('d')

我应该怎么做才能接受这样的输入？

• 输入元素，'XXX'结束：a
• 输入元素，'XXX'结束：b
• 输入元素，'XXX'结束：c
• 输入元素，'XXX'结束：XXX

Answer 1

据我了解，您想获取用户输入并将其附加到堆栈。

您可以使用 list，如下所述：

stack = []
elem = input("Enter element, 'XXX' to end:")

while elem != 'XXX':
    stack.append(elem)
    elem = input("Enter element, 'XXX' to end:")

或者，您也可以使用 collections.deque，它比 list 更有效。

from collections import deque

stack = deque()
elem = input("Enter element, 'XXX' to end:")

while elem != 'XXX':
    stack.append(elem)
    elem = input("Enter element, 'XXX' to end:")

要弹出元素，只需使用 stack.pop() 即可从 list 或 deque.

中移除最后插入的元素