包含min函数的栈
包含min函数的栈
FollowMee 发表于5个月前
包含min函数的栈
  • 发表于 5个月前
  • 阅读 0
  • 收藏 0
  • 点赞 0
  • 评论 0

腾讯云 学生专属云服务套餐 10元起购>>>   

包含min函数的栈

题目内容:定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数

解法:对象中持有两个list,一个用来模拟栈结构保存数据,另外一个保存最小值

  • 常规写法
# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.s1 = []
        self.s2 = []
        
    def push(self, node):
        # write code here
        if not self.s1 and not self.s2: #栈空间为空时插入
            self.s1.append(node)
            self.s2.append(node)
        else:
            self.s1.append(node)
            if node < self.s2[-1]: #插入值小于最小栈顶点值时
                self.s2.append(node)
            else:
                self.s2.append(self.s2[-1]) #插入值大于最小栈顶点值时
    def pop(self):
        # write code here
        if self.s1 and self.s2:
            self.s2.pop()
            return self.s1.pop()
        return None
    def top(self):
        # write code here
        if self.s1:
            return self.s1[-1]
        return None
    def min(self):
        # write code here
        if self.s2:
            return self.s2[-1]
        return None

注意:边界条件

  • Pythonic

直接使用一个list来模拟栈,最小值使用内置min函数得到

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.s = []
    def push(self, node):
        # write code here
        self.s.append(node)
    def pop(self):
        # write code here
        if self.s:
            return self.s.pop()
        return None
    def top(self):
        # write code here
        if self.s:
            return self.s[-1]
        return None
    def min(self):
        # write code here
        if self.s:
            return min(self.s)
        return None

 

标签: Python 算法
共有 人打赏支持
粉丝 2
博文 8
码字总数 2003
×
FollowMee
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: