[每日一题系列]第三天——两个栈实现一个队列的offer()和poll()操作

我的思路

之前在面试里碰到了这道题，当时的第一反应是用第一个栈来负责入队操作，当需要出队的时候，我们知道栈是先进后出，而队列是先进先出，所以我想到的是出队的时候，先把第一个栈的内容压栈到第二个栈里面，从第二个栈里出栈，这样就可以实现先进先出。
对于出队列的时候，我最开始的写法是吧第一个栈的内容压栈到第二个栈里，将第二个栈的栈顶元素出栈，然后再把第二个栈里的元素压回第一个栈，以保证后续对第一个栈压栈不会影响顺序。后来，写到一半，发现其实不用吧第二个栈里的内容再压回去，只要出栈的时候从第二个栈出，并不会影响第一个栈的入栈，当我们第二个栈为空时，就可以再考虑第一个栈里是否还有内容，如果有的话就把他们压到第二个栈里。
当时面试官听了之后，说这是一个方法，但是能不能优化一下，emmmm，我到现在还没想出来怎么进行优化。如果后面有想出来优化办法的话会在后面更新。

我的代码

import java.util.Stack;

public class Queue {
    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> stack2 = new Stack<>();
    
    public boolean offer(int val){
        stack1.push(val); //直接将元素压入stack1
        return true;
    }
    
    public boolean poll(){
        if(stack1.isEmpty() && stack2.isEmpty()){//判断队列是否为空
            return false;
        }
        else{
            if(!stack2.isEmpty()){//如果stack2不空，直接从stack2出栈，即出队操作
                stack2.pop();
            }
            else{//stack2为空，将stack1里的元素压入stack2后，在从stack2出栈，即出队操作
                while(!stack1.isEmpty()){
                    stack2.push(stack1.pop());
                }
                stack2.pop();
            }
            return true;
        }
    }
}