数据结构与算法 | 栈实现 (使用py实现)

用顺序表结构实现

# CY3761 | 2021-12-13 11:32

"""
栈要实现的操作

Stack() 创建一个新的空栈
push(item) 添加一个元素item到栈顶
pop() 弹出栈顶元素
top() 返回栈顶元素
isEmpty() 判断栈是否为空
size() 返回栈的元素个数
"""

# 用顺序表结构实现

class Stack:
    def __init__(self):
        self._items = []
    
    def push(self, *args):  # 进行优化 可以连续添加多个值
        for _ in args:
            self._items.append(_)  # 列表的最后元素作为栈顶
    
    def pop(self):
        if self.isEmpty():  # 这里可以进行抛出异常
            return None
        return self._items.pop()  # 最后元素进行出栈
    
    def top(self):
        if self.isEmpty():  # 这里可以进行抛出异常
            return None
        return self._items[-1]  # 列表的最后元素作为栈顶
    
    def isEmpty(self):
        return self.size() <= 0
    
    def size(self):
        return len(self._items)


if __name__ == '__main__':
    s = Stack()  # 创建一个空栈
    s.push('A', 'B', 'C', 'D', 'E')  # 添加元素
    
    while True:
        pop = s.pop()
        print(pop, s.size())
        
        if not pop:
            break

在这里插入图片描述

用链表结构实现

# CY3761 | 2021-12-13 12:06

"""
栈要实现的操作

Stack() 创建一个新的空栈
push(item) 添加一个元素item到栈顶
pop() 弹出栈顶元素
top() 返回栈顶元素
isEmpty() 判断栈是否为空
size() 返回栈的元素个数
"""

# 用链表结构实现

# 链表有节点类
class Node:
    def __init__(self, _item, _next=None):
        self._item = _item  # 数据域
        self._next = _next  # 指针域
    
    def item(self, *args):
        if not args:
            return self._item
        self._item = args[0]
    
    def next(self, *args):
        if not args:
            return self._next
        _next = args[0]
        
        if isinstance(_next, Node) or _next is None:
            self._next = _next


class Stack:
    def __init__(self):
        self._top = None  # 这里是以链表的第一个节点作为栈顶元素 不断改变
        self._size = 0
    
    def push(self, *args):  # 进行优化 可以连续添加多个值
        """
        _ = A, self._top = Node(A, None), self._size = 1
        _ = B, self._top = Node(B, ANode), self._size = 2
        _ = C, self._top = Node(C, BNode), self._size = 3
        _ = D, self._top = Node(D, CNode), self._size = 4
        _ = E, self._top = Node(E, DNode), self._size = 5
        """
        for _ in args:
            self._top = Node(_, self._top)  # 新建一个节点, 并且将新节点赋值给 _top
            self._size += 1
    
    def pop(self):
        if self.isEmpty():  # 这里可以进行抛出异常
            return None
        self._size -= 1
        
        _oldItem = self._top.item()  # 弹出去的元素
        
        self._top = self._top.next()  # 改变头
        
        return _oldItem
    
    def top(self):
        return self._top  # 这里是以链表的第一个节点作为栈顶元素 不断改变
    
    def isEmpty(self):
        return self.size() <= 0
    
    def size(self):
        return self._size


if __name__ == '__main__':
    s = Stack()  # 创建一个空栈
    s.push('A', 'B', 'C', 'D', 'E')  # 添加元素
    
    """"""
    while True:
        pop = s.pop()
        top = s.top()
        top = top.item() if top else None
        print(pop, s.size(), top)

        if not pop:
            break
    

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

CY3761

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值