Python：基于数组手搓一个栈【算法村E4栈笔记（青铜）】

前言

「栈 Stack」是一种遵循先入后出（First In, Last Out）原则的线性数据结构。

我们可以将栈类比为桌面上的一摞盘子，如果需要拿出底部的盘子，则需要先将上面的盘子依次取出。我们将盘子替换为各种类型的元素（如整数、字符、对象等），就得到了栈数据结构。

在栈中，我们把堆叠元素的顶部称为「栈顶」，底部称为「栈底」。将把元素添加到栈顶的操作叫做「入栈」，而删除栈顶元素的操作叫做「出栈」。

（以上文字来自：《Hello 算法》）

---

栈的基本功能

# 初始化栈
# Python 没有内置的栈类，可以把 List 当作栈来使用 
stack: List[int] = []

# 元素入栈
stack.append(1)
stack.append(3)
stack.append(2)
stack.append(5)
stack.append(4)

# 访问栈顶元素
peek: int = stack[-1]

# 元素出栈
pop: int = stack.pop()

# 获取栈的长度
size: int = len(stack)

# 判断是否为空
is_empty: bool = len(stack) == 0

---

基于数组，实现一个栈

# 以下：基于数组实现栈
class Stack(object):
    """栈"""

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

    def is_empty(self):
        """判断是否为空"""
        return self.items == []

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

    def pop(self):
        """弹出元素"""
        return self.items.pop()

    def peek(self):
        """返回站定元素"""
        return self.items[-1]

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

测试：

stack = Stack()

stack.push(1)
stack.push(2)
stack.push(3)

print(stack.peek())  # 3
print(stack.pop())  # 3
print(stack.pop())  # 2