栈是一种基本数据结构,它的特点是先入后出,我们在选用python实现栈时,首先需要考虑到必须要选用可以更改的数据类型。因此,我们选择最常用的列表。这里不得不提一句,列表真是yyds,虽然很多情况下列表会增加算法的复杂度,但对新人简直太友好了。
我们可以想象一下,最先存入栈中的元素,即最先存入列表中的元素,应该是列表中索引为0的第一个元素,后面依次存入元素,最后存入栈中的元素就是列表最后一个元素。也就是说,索引为0表示栈底,索引为len(列表名)-1表示栈底,当然直接用-1也可以。
我们选择移除栈中元素时,因为栈只能先入后出,所以我们每次只要移除列表最后一个元素就行,正好pop()函数的复杂度是1,这就很nice了。
代码如下:
class Stack:
def __init__(self):
#我们定义空列表,实现栈的进入进出操作
self.items = []
def isEmpty(self):
#判断是否为空栈
return self.items == []
def push(self,item):
#向栈中加入元素,新的元素自动加入栈底
self.items.append(item)
def pop(self):
#移出栈顶元素。
return self.items.pop()
def peek(self):
#返回栈顶元素
return self.items[len(self.items)-1]
def size(self):
#返回栈的长度
return len(self.items)