Python - 数据结构之栈 (stack)

  • 简述

    在英语词典中,stack 一词意味着将对象排列在另一个之上。在此数据结构中分配内存的方式相同。它以类似的方式存储数据元素,就像一堆盘子在厨房里一个接一个地存储一样。因此堆栈数据结构允许在一端进行操作,这可以称为堆栈顶部。我们只能从堆栈中添加元素或删除元素。
    在堆栈中,按顺序最后插入的元素将首先出现,因为我们只能从堆栈顶部移除。这种功能称为后进先出(LIFO)功能。添加和删除元素的操作称为PUSHPOP. 在下面的程序中,我们将其实现为add并且和remove职能。我们声明一个空列表并使用 append() 和 pop() 方法来添加和删除数据元素。
  • 推入堆栈

    让我们了解一下,如何在 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())
    

    输出

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