欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

面试之手撕代码二

程序员文章站 2024-03-21 22:05:28
...

(二)
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
解法:关键是如何在常数时间做到,解法为双栈(空间换时间),一个存数据,一个存最小值,push的同时,也在第二个栈push最小值(或者push的时候只比较当前值与最小栈的栈顶,不更大push,pop的同时判断是否与第二个栈顶相同,同则pop)。
Python版(第一种解法):

class MinStack(object):

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.result = []
        self.min = []

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.result.append(x)
        if len(self.min) == 0:
            self.min.append(x)
        else:
            if x < self.min[-1]:
                self.min.append(x)
            else:
                self.min.append(self.min[-1])
    def pop(self):
        """
        :rtype: None
        """
        self.min.pop()
        return self.result.pop()
    def top(self):
        """
        :rtype: int
        """
        return self.result[-1]

    def getMin(self):
        """
        :rtype: int
        """
        if len(self.min) == 0:
            return []
        return self.min[-1]