题解 | #设计getMin功能的栈#

设计getMin功能的栈

http://www.nowcoder.com/practice/05e57ce2cd8e4a1eae8c3b0a7e9886be

class Stack:
    def __init__(self):
        self.A = []
        self.B = []
    def push(self, val):
        self.A.append(val)
        if not self.B or self.B[-1] >= val:
            self.B.append(val)
    def pop(self):
        if self.B and self.A.pop() == self.B[-1]:
            self.B.pop()
    def getMin(self):
        if self.B:
            return self.B[-1]
        return -1

s = Stack()
n = int(input())
for _ in range(n):
    l = input().split()
    if l[0] == 'push':
        s.push(int(l[1]))
    elif l[0] == 'pop':
        s.pop()
    elif l[0] == 'getMin':
        print(s.getMin())       
全部评论

相关推荐

04-28 11:34
西北大学 运营
牛客4396号:不好意思,这个照片猛一看像丁真
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务