剑指offer-005-用两个栈实现队列 -- Java实现

题目

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

分析

思路一：

栈：先进后出，一般从尾部添加，也从尾部删除。
队列：先进先出，从尾部添加，从头部删除。
使用两个栈（A与B），A栈负责push，B栈负责pop。A栈pop出的元素push进B栈。队列即从A栈push进，从B栈pop出，这样，经过两个栈“负负得正”即可实现先进先出。
细节：在实现pop时，应注意如果B栈为空，则需要判断A栈是否也为空，如果不为空，则需将A栈的元素全部先push进B栈。然后最后在pop出元素。

push时间复杂度：o(1)
pop空间复杂度：o(1)

代码：

import java.util.Stack;

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