Python教程：堆栈数据操作

在英语词典中，堆栈（Stack）的意思是将一个对象放在另一个对象之上。该数据结构中分配内存的方式是相同的。它存储数据元素的方式与厨房中一堆盘子类似：一个放在另一个上面。因此，启用操作的堆栈数据的末尾可以称为堆栈顶部。元素可以添加到堆栈顶部 或简单地从堆栈中删除。

在堆栈中，序列的最后一个元素首先出现，因为它只能从堆栈顶部删除。此函数称为后进先出 (LIFO)。添加和删​​除项目的操作是PUSH和POP。在下面的程序中，我们将其实现为函数 add 和 remove。首先，声明一个空列表并使用 append() 和 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 peek to look at the top of the stack

    def peek(self):     
        return self.stack[0]

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

Python

运行上面的示例代码，得到以下结果：

Mon
Mon

Shell

堆栈移除

只能从堆栈中移除数据项。下面是一个可以实现此功能的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")
print(AStack.remove())
AStack.add("Wed")
AStack.add("Thu")
print(AStack.remove())

Python

运行上面的示例代码，得到以下结果 -

Tue
Thu