剑指offer 第5题,用两个栈实现队列

描述:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型

思路:

1. 准备两个栈,stack1, stack2

2. push时,stack1 push

3. pop时,如果stack2不为空,则stack2pop,否则,stack1依次pop出所有元素,append到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 self.stack2:
            return self.stack2.pop()
        else:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
            return self.stack2.pop()

结果:

运行时间:25ms

占用内存:5736k

猜你喜欢

转载自blog.csdn.net/qq_38650545/article/details/86177236