Python - 堆栈的数据结构



在英语词典中,单词 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 函数返回最顶层的元素。我们首先通过计算堆栈的大小来检查 top 元素,然后使用内置的 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