剑指offer 第二版(Python3)--面试题9:用两个栈实现队列,用两个队列实现栈

第2章 面试需要的基础知识

  面试题4:二维数组中的查找
  面试题5:替换空格
  面试题6:从尾到头打印链表
  面试题7:重建二叉树
  面试题9:用两个栈实现队列

第3章 高质量的代码

第4章 解决面试题的思路

第5章 优化时间和空间效率

第6章 面试中的各项能力

第7章 两个面试案例

题目描述
牛客网 用两个栈实现队列
  用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

解题思路
牛客网

  队列的特性是:“先入先出”,栈的特性是:“先入后出”。我们向模拟的队列插入数 a,b,c 时,假设插入的是 stack1,此时的栈情况为:

  栈 stack1:{a,b,c}
  栈 stack2:{}

  当需要弹出一个数,根据队列的"先进先出"原则,a 先进入,则 a 应该先弹出。但是此时 a 在 stack1 的最下面,将 stack1 中全部元素逐个弹出压入 stack2,然后从 stack2 中弹出 a,此时的栈情况为:

  栈 stack1:{}
  栈 stack2:{c,b}

  继续弹出一个数,b 比 c 先进入"队列",b 弹出,注意此时 b 在 stack2 的栈顶,可直接弹出,此时的栈情况为:

  栈 stack1:{}
  栈 stack2:{c}

  此时向模拟队列插入一个数 d,还是插入 stack1,此时的栈情况为:

  栈 stack1:{d}
  栈 stack2:{c}

  弹出一个数,c 比 d 先进入,c 弹出,注意此时 c 在 stack2 的栈顶,可直接弹出,此时的栈情况为:

  栈 stack1:{d}
  栈 stack2:{}

  总结:

  插入时,直接插入到 stack1;
  弹出时,若 stack2 不为空,弹出 stack2 栈顶元素,如若stack2 为空,将 stack1 中的全部数逐个出栈,并入栈 stack2,再弹出 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 not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        if self.stack2:
            return self.stack2.pop()
        return None

相关题目
  用两个队列实现一个栈

解题思路
  这个题比上一题稍微复杂一点。
  我们向模拟的队列插入数 a ,假设插入的是 queue1,此时的栈情况为:

  栈 queue1:{a}
  栈 queue2:{}

  继续插入数b,加入到queue2,并将queue1中数弹出到queue2中,此时队列情况为:

  栈 queue1:{}
  栈 queue2:{b, a}

  继续插入数c,加入到queue1,并将queue2中数弹出到queue1中,此时队列情况为:

  栈 queue1:{c, b, a}
  栈 queue2:{}

  然后执行一次弹出操作,从非空队列中弹出一个数,这里从queue1中弹出c,此时队列情况为:

  栈 queue1:{b, a}
  栈 queue2:{}

  然后插入数d,加入到queue2,并将queue1中数全部弹出到queue2中,此时队列情况为:

  栈 queue1:{}
  栈 queue2:{d, b, a}

  然后执行一次弹出操作,从非空队列中弹出一个数,这里从queue2中弹出d,此时队列情况为:

  栈 queue1:{}
  栈 queue2:{b, a}

  总结,插入操作是插入到非空队列中,并将另一个非空队列中的数全部弹出到该队列中,这样保持一个队列为空;弹出操作是从非空队列弹出一个数。

实战

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

	def push(self, num):
		if not self.queue1:
			self.queue1.append(num)
			while self.queue2:
				self.queue1.append(self.queue2.pop(0))
		else:
			self.queue2.append(num)
			while self.queue1:
				self.queue2.append(self.queue1.pop(0))

	def pop(self,):
		if self.queue1:
			return self.queue1.pop(0)
		if self.queue2:
			return self.queue2.pop(0)
		return None
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值