直通BAT面试算法精讲--队列和栈（2）

案例二

双栈队列练习题
题干：
编写一个类，只能用两个栈结构实现队列，支持队列的基本操作（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