Python-堆栈


在英语词典中,“堆栈”一词的意思是将对象排列在另一个对象之上。这与在此数据结构中分配内存的方式相同。它以类似于厨房中一堆盘子叠放在一起的方式存储数据元素。因此,堆栈数据结构允许在一端(称为堆栈顶部)进行操作。我们只能从堆栈的这一端添加元素或删除元素。

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

压入堆栈

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

从堆栈中弹出

正如我们所知,我们只能从堆栈中删除最顶层的数据元素,我们实现了一个 python 程序来执行此操作。以下程序中的删除函数返回最上面的元素。我们首先通过计算堆栈的大小来检查顶部元素,然后使用内置的 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())

输出

执行上述代码时,会产生以下结果 -

Thu
Wed