案例二
双栈队列练习题
题干:
编写一个类,只能用两个栈结构实现队列,支持队列的基本操作(add,poll,peek)
解题思路:
两个栈,一个用来实现入队,一个用来实现出队
class Solution():
def __init__(self):
self.stack1 = []
self.stack2 = []
def push(self,number):
self.stack1.append(number)
def pop(self):
if len(self.stack1)==0 and len(self.stack2)==0:
return
elif len(self.stack2)==0:
while len(self.stack1)>0:
self.stack2.append(self.stack1.pop())
return self.stack2.pop()
拓展:
#用两个队列实现栈的操作
class Solution():
def __init__(self):
self.queue1 = []
self.queue2 = []
def push(self,number):
self.queue1.append(number)
def pop(self):
if len(self.queue1)==0:
return
while len(self.queue1)!=1:
self.queue2.append(self.queue1.pop(0))
self.queue1,self.queue2 = self.queue2,self.queue1
return self.queue2.pop()
案例三
栈的翻转练习题
题干:
实现一个栈的逆序,但是只能用递归函数和这个栈本身的操作来实现,而不能自己申请另外的数据结构
测试样例:
[4,3,2,1]
返回:[1,2,3,4]
解题思路:
递归地弹出栈底元素
递归地将栈底元素入栈
class Solution():
def reverse(self,stack):
if len(stack)==0:
return
i = self.get(stack)
self.reverse(stack)
stack.append(i)
return stack
def get(self,stack):
result = stack.pop()
if len(stack)==0:
return result
else:
last = self.get(stack)
stack.append(result)
return last
案例四
双栈排序练习题
一个栈中元素类型为整型,现在想将该栈从顶到底按照从大到小的排序,要求最多只能申请一个额外的栈存放临时数据,除此之外可以申请新的变量,但不能申请额外的数据结构,如何完成排序?
测试样例:
输入: [2,1,4,5,3]
返回[1,2,3,4,5]
解题思路:
申请个辅助栈help,从stack栈中弹出栈顶元素current,如果current小于help的栈顶元素,则入栈help。否则,将help中的元素逐渐弹出,并且重新压回stack中,直到current小于等于help的栈顶元素,将current入栈help,直到stack中的所有元素都入栈help,再将help中的所有元素出栈入栈至stack
class Solution():
def sort(self,stack):
if len(stack) == 0:
return
help = []
while len(stack)>0:
current = stack.pop()
while len(help)>0 and current>help[-1]:
stack.append(help.pop())
help.append(current)
while len(help)>0:
stack.append(help.pop())
return stack