> 文档中心 > 用python实现栈

用python实现栈

栈是一种基本数据结构,它的特点是先入后出,我们在选用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)