栈
栈(stack)是一种容器,它的特点在于只允许在容器的一端(称为栈顶指标,即top)进行加入数据(push)和输出数据(pop)的运算,即按照后进先出(LIFO,Last In First Out)的原理运作。
栈可以用顺序表实现,也可以用链表实现。顺序表或链表决定了数据如何存放,栈决定了数据如何操作。
Python中的列表(List)就相当于顺序表,现在我们采用List来实现栈。
我们需要实现的栈包含以下几种操作:
- Stack() 创建一个新的空栈
- push(elem) 添加一个新的元素elem到栈顶
- pop() 弹出栈顶元素
- peek() 返回栈顶元素
- is_empty() 判断栈是否为空
- size() 返回栈的元素个数
下面直接看代码实现:
class Stack(object):
"""栈"""
def __init__(self):
self.__list = []
def push(self, elem):
"""添加一个新元素elem到栈顶"""
self.__list.append(elem)
def pop(self):
"""弹出栈顶元素"""
if self.is_empty():
return None
else:
return self.__list.pop()
def peek(self):
"""返回栈顶元素"""
if self.__list:
return self.__list[-1]
else:
return None
def is_empty(self):