剑指offer：用两个栈实现队列

题目描述

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

 

解题思路

入队：将元素进栈A 
出队：判断栈B是否为空，如果为空，则将栈A中所有元素pop，并push进栈B，栈B出栈； 
如果不为空，栈B直接出栈。 
代码如下：

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(stack1.empty()&&stack2.empty()){ 
            throw new RuntimeException("Queue is empty!"); 
        } 
        if(stack2.empty()){ 
            while(!stack1.empty()){ 
                stack2.push(stack1.pop()); 
            } 
        } 
        return stack2.pop(); 
    } 
} 

 

拓展（用队列模拟栈）

入栈：始终保持一个队列为空，将新插入的元素加入到空队列中，再将不为空的队列依次取出放到另一个队列中。 
出栈：弹出不为空的那一个队列。