Python数据结构堆栈

在英语词典中，堆栈这个词的意思是将对象排列在另一个上面。在这个数据结构中，内存的分配方式也是如此。它存储数据元素的方式类似于在厨房里把一堆盘子一个一个地存放起来。因此，堆栈数据结构允许在一端进行操作，这一端可以称为堆栈的顶部。我们只能在堆栈的这一端添加元素或删除元素。

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

向堆栈中PUSH

让我们了解一下，如何在堆栈中使用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())

输出

当上述代码被执行时，它产生了以下结果 –