剑指offer 7题 【栈和队列】用两个栈实现队列

题目描述

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

牛客传送门：点击打开链接

public class Title7 {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.add(node);
    }
    
    public int pop() {
        if(!stack2.empty()){
            return stack2.pop();
        }else{
            // 如果俩个栈全为空，返回0
            if(stack1.empty())
                return 0;
            while(!stack1.empty()){
                stack2.add(stack1.pop());
            }
            return stack2.pop();
        }
    }
    
    public static void main(String[] args) {
        Title7 t = new Title7();
        t.push(1);
        t.push(2);
        System.out.println(t.pop());
        t.push(3);
        System.out.println(t.pop());
        t.push(4);
        System.out.println(t.pop());
        System.out.println(t.pop());
    }
}