数据结构之栈python版

:是一种仅在表尾进行插入和删除操作的线性表,常称为压栈和出栈。借用网上的一个段子很形象:吃多了拉就是队列,吃多了吐就是栈。

python代码实现栈如下:

# coding:utf-8

class Stack(object):
    """
    栈:后进先出(LIFO)
    """

    def __init__(self):
        self._list = []

    def push(self, item):
        """
        压栈:添加一个元素到栈顶
        """
        self._list.append(item)

    def pop(self):
        """
        出栈:弹出栈顶元素
        """
        return self._list.pop()

    def peek(self):
        """
        返回栈顶元素
        """
        return self._list[-1] if not self.is_empty() else None

    def is_empty(self):
        """
        判断栈是否为空
        """
        return not self._list

    def size(self):
        """
        返回栈的大小
        """
        return len(self._list)

if __name__ == '__main__':
    s = Stack()
    print(s.is_empty())
    s.push(1)
    s.push(2)
    s.push(3)
    print(s.size())
    print(s.pop())
    print(s.pop())
    print(s.is_empty())

你可能感兴趣的:(数据结构之栈python版)