9:用两个栈实现队列(拓展:用两个队列实现一个栈)

用两个栈来实现一个队列,声明如下:实现队列的Push和Pop操作,即在队尾插入节点和队头删除插入节点。 队列中的元素为int类型。

class Queue:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
        # self.stack1 = [x for x in list]
        # self.stack2 = []这种方式错误,并不是一开始就给出一个队列,而是自己组建一个队列并实现其功能

    def push(self, node):
        return self.stack1.append(node)
        # 因为是队列所以不能用切片操作,因为体现不出过程
        # self.stack1.append(x)
        # self.stack2 = self.stack1[::-1]

    def pop(self):
        if self.stack2:
            return self.stack2.pop()
        elif not self.stack1:
            return None
        else:
            self.stack2.append(self.stack1.pop())
            return self.stack2.pop()
        # self.stack1.pop()
        # self.stack2 = self.stack1[::-1]
        # return self.stack2

题目拓展:用两个队列实现一个栈

class Solution:
    def __init__(self):
        self.queue1 = []
        self.queue2 = []

    def push(self, node):
        return self.queue1.append(node)

    def pop(self):
        if not self.queue1:
            return None
        else:
            while len(self.queue1)!= 1:
                self.queue2 =self.queue1.pop()
            self.queue1,self.queue2 =self.queue2,self.queue1
            #这一步交换不能少,否则就只能执行一次pop操作
            return self.queue2.pop()
# pycharm alt+j是选中多个单词
发布了16 篇原创文章 · 获赞 0 · 访问量 155

猜你喜欢

转载自blog.csdn.net/qq_43275748/article/details/102774994
今日推荐