用两个栈实现队列

题目描述

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

 

分析：

可以用两个栈来实现，stack1和stack2。先将元素逐个压入stack1中，如果stack2为空，则把stack1中的元素逐个弹出压入stack2中，由于先进入队列的元素被压到stack1的底端，之后的弹出和再压入stack2中，这样原来stack1底端的元素，就变成了stack2顶端的元素了，再进行弹出，这样就实现了队列的操作了。如果stack2不为空，在stack2中栈顶元素是最先进入队列的元素，可以弹出。

 

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());//把stack1的值弹出后再压入stack2中
            }
        }
        return stack2.pop();//弹出
    }
}