栈和自定义异常
#!/usr/bin/env class StackFullError(Exception): """ 自定义异常 """ pass class StackNullError(Exception): """ 自定义异常 """ pass class Stack: """ 栈:先进后出 比如用笼屉蒸包子,最底下一层是最先放的,但是最后拿出来的。 """ def __init__(self, size): """ 实例化对象 :param size: 栈大小 """ self.size = size self.top = 0 # 栈指针位置 self.lst = [] # 用list做栈 def push(self, val): """ 往栈里面添加值 1. 往栈指针所在位置添加值 2. 添加成功以后,栈指针+1 :param val: :return: null """ if self.top == self.size: raise StackFullError("stack in full!!!") self.lst.insert(self.top, val) self.top += 1 def get(self): """ 取栈里面的值,按栈顺序取 1. 指针位置-1 获取到有值的栈位置 2. 返回获取到值 :return: val """ if self.top == 0: raise StackNullError("stack in null!!!") self.top -= 1 return self.lst[self.top] obj = Stack(5) obj.push("A") obj.push("B") obj.push("C") obj.push("D") obj.push("E") print(obj.get()) print(obj.get()) print(obj.get()) print(obj.get()) print(obj.get()) print(obj.get())