Python实现数据结构——栈 Python

# 创建一个空栈（列表）
stack = []

# 压入元素
stack.append(1)
stack.append(2)
stack.append(3)

# 弹出栈顶元素
top = stack.pop()
print(top)   # 3

# 查看栈顶元素（不弹出）
print(stack[-1])  # 2

# 判断栈是否为空
print(len(stack) == 0)  # False

class Stack:
    """栈的实现"""
    def __init__(self):
        self.items = []   # 用列表存储元素

    def push(self, item):
        """压入元素"""
        self.items.append(item)

    def pop(self):
        """弹出并返回栈顶元素"""
        if self.is_empty():
            raise IndexError("栈为空，无法弹出")
        return self.items.pop()

    def peek(self):
        """查看栈顶元素（不弹出）"""
        if self.is_empty():
            return None
        return self.items[-1]

    def is_empty(self):
        """判断栈是否为空"""
        return len(self.items) == 0

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

# 实例演示
s = Stack()
s.push(5)
s.push(10)
s.push(15)
print(s.pop())   # 15
print(s.peek())  # 10
print(s.size())  # 2