Python - 堆栈

更新于 2024/1/29 1:13:00

在英语词典中，单词 stack 的意思是将对象排列在另一个之上。这与在此数据结构中分配内存的方式相同。它以类似的方式存储数据元素，就像一堆盘子在厨房里一个叠一个地存储一样。所以堆栈数据结构允许在可以称为堆栈顶部的一端进行操作。我们只能从堆栈的这一端添加元素或移除元素。

在堆栈中，顺序最后插入的元素将首先出现，因为我们只能从堆栈顶部移除。这种功能被称为后进先出（LIFO）功能。添加和删除元素的操作称为PUSH 和POP。在下面的程序中，我们将其实现为 add 和 remove 函数。我们声明一个空列表并使用 append() 和 pop() 方法添加和删除数据元素。

堆栈中的 PUSH 入栈

让我们了解如何在 Stack 中使用 PUSH。请参阅下面提到的程序 −

示例

class Stack:
   def __init__(self):
      self.stack = []

   def add(self, dataval):
# Use list append method to add element
      if dataval not in self.stack:
         self.stack.append(dataval)
         return True
      else:
         return False
# Use peek to look at the top of the stack
   def peek(self):     
	   return self.stack[-1]

AStack = Stack()
AStack.add("Mon")
AStack.add("Tue")
AStack.peek()
print(AStack.peek())
AStack.add("Wed")
AStack.add("Thu")
print(AStack.peek())

输出

当上面的代码执行时，会产生如下结果 −

Tue
Thu

堆栈中的 POP 出栈

正如我们所知，我们只能从堆栈中删除最顶部的数据元素，我们实现了一个执行此操作的 python 程序。以下程序中的 remove 函数返回最顶层的元素。我们首先通过计算堆栈的大小来检查顶部元素，然后使用内置的 pop() 方法找出最顶部的元素。

class Stack:
   def __init__(self):
      self.stack = []

   def add(self, dataval):
# Use list append method to add element
      if dataval not in self.stack:
         self.stack.append(dataval)
         return True
      else:
         return False
        
# Use list pop method to remove element
   def remove(self):
      if len(self.stack) <= 0:
         return ("No element in the Stack")
      else:
         return self.stack.pop()

AStack = Stack()
AStack.add("Mon")
AStack.add("Tue")
AStack.add("Wed")
AStack.add("Thu")
print(AStack.remove())
print(AStack.remove())