python两个队列实现一个栈和两个栈实现一个队列

1.两个栈实现一个队列

两个栈stack1和stack2， push的时候直接push进stack1，pop时需要判断stack1和stack2中的情况。如果stack2不为空的话，直接从stack2中pop，如果stack2为空，把stack1中的值push到stack2中，然后再pop stack2中的值。

class Solution:

    def __init__(self):

        self.stack1 = []

        self.stack2 = []

    def push(self, node):

        # write code here

        self.stack1.append(node)

    def pop(self):

        # return xx

        if len(self.stack1) == 0 and len(self.stack2) == 0:

            return

        elif len(self.stack2) == 0:

            while len(self.stack1) > 0:

                self.stack2.append(self.stack1.pop())

        return self.stack2.pop()

2.两个队列实现一个栈

进栈：元素入队列A

出栈：判断如果队列A只有一个元素，则直接出队。否则，把队A中的元素出队并入队B，直到队A中只有一个元素，再直接出队。为了下一次继续操作，互换队A和队B。

class Stock:

    def __init__(self):

        self.queueA=[]

        self.queueB=[]

    def push(self, node):

        self.queueA.append(node)

    def pop(self):

        if len(self.queueA)==0:

            return None

        while len(self.queueA)!=1:

            self.queueB.append(self.queueA.pop(0))

        self.queueA,self.queueB=self.queueB,self.queueA #交换是为了下一次的pop

        return self.queueB.pop()

巴特西

python两个队列实现一个栈和两个栈实现一个队列

最新文章

热门文章