栈和自定义异常

#!/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())

 

转载于:https://www.cnblogs.com/spf21/p/10429368.html