leetcode-辅助栈

1、用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

思路：

一个栈A用来添加,一个b用来删除

删除的时候判断b是否为空，为空需要将a中的数据反向加入B中，返回pop掉b中的最后一个

class CQueue:

    def __init__(self):
        self.a,self.b=[],[]

    def appendTail(self, value: int) -> None:
        self.a.append(value)

    def deleteHead(self) -> int:
        if(self.b):
            return self.b.pop()
        else:
            while(self.a):
                self.b.append(self.a.pop())
            if(not self.b):
                return -1
            else:
                 return self.b.pop()

2、

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

思路：维护一个非严格降序的栈B，这个栈B最后一位为最小值。

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """

        self.a,self.b=[],[]
    def push(self, x: int) -> None:
        self.a.append(x)
        if(not self.b):
            self.b.append(x)
        elif(self.b[-1]>=x):
            self.b.append(x)
    def pop(self) -> None:        
        if(self.a.pop() == self.b[-1]):
            self.b.pop()

    def top(self) -> int:
        return self.a[-1]

    def min(self) -> int:
        return self.b[-1]