两个栈实现队列

题目

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

我的解法

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        while(!stack1.isEmpty()){
            stack2.add(stack1.pop());
        }
        stack2.add(node);
    }
    
    public int pop() {
    	while(!stack2.isEmpty()){
            stack1.add(stack2.pop());
        }
        return stack1.pop();
    }
}

算法分析：push时就把stack1全部压入stack2中，再add到stack2中；pop时就把stack2全部压入stack1中，再pop()stack1。时间复杂度都是O（n）

答案解法

import java.util.Stack;
import java.lang.Exception;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    //stack1入栈，保存进入顺序
    public void push(int node) {
        stack1.add(node);
    }
    
    
    //stack2出栈，保存出去顺序。当stack2为空时，stack1中数据再过来
    public int pop() throws Exception{
    	if(!stack2.isEmpty())
            return stack2.pop();
        else{
            while(!stack1.isEmpty())
                stack2.add(stack1.pop());
        }
        if(stack2.isEmpty())
            throw new Exception("queue is empty");
        return stack2.pop();
    }
}

算法分析：不用把数据在stack1和stack2之间倒来倒去。时间复杂度为o(lg(n))。