包含min函数的栈

包含min函数的栈_牛客网

https://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49?tpId=13&tqId=11173&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

主流思路是应用于一个辅助栈,也就是最小元素栈。 每次压栈操作时, 如果压栈元素比当前最小元素更小, 就把这个元素压入最小元素栈, 原本的最小元素就成了次小元素. 同理, 弹栈时, 如果弹出的元素和最小元素栈的栈顶元素相等, 就把最小元素的栈顶弹出.
在这里我是用了一个技巧,每次压栈时压入一个二元组,包括需要压入的元素和当前的最小元素。

def __init__(self):
    self.q = []
def push(self, node):
    # write code here
    curMin = self.min()
    if curMin == None or node < curMin:
        curMin = node
    self.q.append((node, curMin))

def pop(self):
    # write code here
    if len(self.q) == 0:
        return None
    else:
        return self.q.pop()

def top(self):
    # write code here
    if len(self.q) == 0:
        return None
    else:
        return self.q[len(self.q) - 1][0]

def min(self):
    # write code here
    if len(self.q) == 0:
        return None
    else:
        return self.q[len(self.q) - 1][1]
全部评论

相关推荐

牛客868257804号:九个中铁八个中建
点赞 评论 收藏
分享
头像
11-18 16:08
福州大学 Java
影流之主:干10年不被裁,我就能拿别人一年的钱了,日子有盼头了
点赞 评论 收藏
分享
12 收藏 评论
分享
牛客网
牛客企业服务