2021-02-26 栈和队列

栈

class Stack(object):
    def __init__(self):
        self.__list = []

    def push(self, item):
        # 添加一个新的元素item到栈顶
        self.__list.append(item)

    def pop(self):
        # 弹出栈顶元素
        return self.__list.pop()

    def peek(self):
        # 返回栈顶元素
        if self.__list:
            return self.__list[-1]
        else:
            return None

    def is_empty(self):
        # 判断栈是否为空
        return self.__list == []

    def size(self):
        # 返回栈的元素个数
        return len(self.__list)


if __name__ == "__main__":
    stack = Stack()
    stack.push("hello")
    stack.push("world")
    stack.push("it cast")

队列

class Queue(object):
    def __init__(self):
        self.__list = []

    def enqueue(self, item):
        # 往队列中添加一个item元素
        # self.__list.append(item)
        self.__list.insert(0, item)
        # 有两个方向 取决于那个操作更频繁

    def dequeue(self):
        # 从队列头部删除一个元素
        # self.__list.pop(0)
        return self.__list.pop()

    def is_empty(self):
        # 判断一个队列是否为空
        return self.__list == []

    def size(self):
        # 返回队列的大小
        return len(self.__list)


if __name__ == "__main__":
    q = Queue()
    q.enqueue("hello")
    q.enqueue("world")
    q.enqueue("it cast")
    print(q.size())
    print(q.dequeue())
    print(q.dequeue())
    print(q.dequeue())

class DoubleQueue(object):
    def __init__(self):
        self.__list = []

    def add_front(self, item):
        # 从队头加入一个item元素
        self.__list.insert(0, item)

    def add_rear(self, item):
        # 从队尾加入一个item元素
        self.__list.append(item)

    def remove_front(self):
        # 从队头删除一个item元素
        return self.__list.pop(0)

    def remove_rear(self):
        # 从队尾删除一个item元素
        return self.__list.pop()

    def is_empty(self):
        # 判断一个队列是否为空
        return self.__list == []

    def size(self):
        # 返回队列的大小
        return len(self.__list)


if __name__ == "__main__":
    deque = DoubleQueue()
    deque.add_front(1)
    deque.add_front(2)
    deque.add_rear(3)
    deque.add_rear(4)
    print(deque.size())
    print(deque.remove_front())
    print(deque.remove_front())
    print(deque.remove_rear())
    print(deque.remove_rear())