用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)