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

你可能感兴趣的:(数据结构,python)