【数据结构专项】链表(3) —— 用单链表实现栈(Python)

原创

Changxing长行 2021-08-26 10:48:42 博主文章分类：数据结构与算法专项 ©著作权

©著作权归作者所有：来自51CTO博客作者Changxing长行的原创作品，请联系作者获取转载授权，否则将追究法律责任

用单链表实现栈

栈顶位于单链表的头结点处，维护头指针和链表长度；每次压入元素则在头结点前插入新结点，取出元素则删除头结点。

class LinkedStack:
    """单链表实现的栈"""

    class _Node:
        """"""
        __slots__ = "value", "next"  # 因为会创建很多个结点对象，因此使用__slots__来减少内存占用

        def __init__(self, value, next):
            self.value = value
            self.next = next

    def __init__(self):
        self._head = None  # 头指针
        self._size = 0  # 栈中元素的数量

    def __len__(self):
        """返回栈中元素的数量"""
        return self._size

    def is_empty(self):
        """返回栈是否为空"""
        return self._size == 0

    def push(self, value):
        """向栈中压入元素"""
        self._head = self._Node(value, self._head)  # 构造链表对象
        self._size += 1

    def top(self):
        """查询栈顶元素"""
        if self.is_empty():
            raise ValueError("Stack is Empty")
        return self._head.value

    def pop(self):
        """取出栈顶元素"""
        if self.is_empty():
            raise ValueError("Stack is Empty")
        ans = self._head.value
        self._head = self._head.next
        self._size -= 1
        return ans