Python数据结构与算法之 栈结构的实现

压栈（入栈）：push(item) 添加一个新的元素item到栈顶。
弹出（出栈）：pop() 弹出栈顶元素。
peek() 返回栈顶元素。
is_empty() 判断栈是否为空
size() 返回栈的元素个数。

# -*- coding: UTF-8 -*-
class Stack(object):
    '''栈'''
    def __init__(self):
        self.__list = []

    def push(self, item):
        '''添加一个新元素item到栈顶'''
        self.__list.append(item) #O(1)
        #self.__list.insert(0, item) #顺序表存储复杂度高。O(n)

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

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

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

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

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

运行结果：
3
2
1