第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