JZ5 - 用两个栈实现队列

原创于 2020-06-08 12:17:20 发布 · 177 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用两个栈来实现队列的方法，详细解释了如何通过栈1进行入队操作，栈2进行出队操作，当栈2为空时，栈1的所有元素将移至栈2，再进行出队操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

解题思路

栈1 用来作入队列
栈2 用来出队列，当栈2为空时，栈1全部出栈到栈2，栈2再出栈（即出队列）

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.st1 = []
        self.st2 = []
    def push(self, node):
        # write code here
        self.st1.append(node)
    def pop(self):
        # return xx
        if self.st2==[]:
            for i in range(len(self.st1)):
                self.st2.append(self.st1.pop())
        return self.st2.pop()