题目

用栈实现队列

正如标题所述,你需要使用两个栈来实现队列的一些操作。

队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例

比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2

挑战

仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的

解题

两个栈stack1 、stack2,一个存储入队列元素,一个存储出出队列元素

stack2 入队列

stack1出队列

当入队列的时候:直接在入栈stack2

当出队列的时候:若栈stack2中有元素,则栈stack2底的元素就是所要出队列的元素,将栈stack2中的元素全部出来,并放到栈stack1中,此处stack1栈顶元素就是答案

这里有个问题是若栈stack1中有元素,则就直接取出栈stack1的栈顶元素就是答案

Java

public class Queue {
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;

    public Queue() {
       // do initialization if necessary
       stack1 = new Stack<Integer>();
       stack2 = new Stack<Integer>();
    }
    private void stack2Tostack1(){
        while(!stack2.empty()){
            stack1.push(stack2.peek());
            stack2.pop();
        }
    }
    public void push(int element) {
        // write your code here
        stack2.push(element);
        
    }

    public int pop() {
        // write your code here
        if( stack1.empty()){
            this.stack2Tostack1();
        }
        return stack1.pop();
    }

    public int top() {
        // write your code here
        if(stack1.empty()){
            this.stack2Tostack1();
        }
        return stack1.peek();
    }
}
Java Code

Python

class MyQueue:

    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    
    def push(self, element):
        # write your code here
        self.stack2.append(element)

    def top(self):
        # write your code here
        # return the top element
        self.stack2Tostack1()
        return self.stack1[len(self.stack1)-1]

    def pop(self):
        # write your code here
        # pop and return the top element
        self.stack2Tostack1()
        return self.stack1.pop()
    
    def stack2Tostack1(self):
        if len(self.stack1)==0:
            while len(self.stack2)!=0:
                self.stack1.append(self.stack2.pop())
Python Code

参考了九章 ,由于初始化不知道怎么定义