双栈实现队列

1|0俩个栈实现队列的功能，头部删除，尾部添加，是否为空，返回队头

/**
 * 用俩个栈实现队列
 */
public class MyQueue {

    Stack<Integer> stack1,stack2;

    public MyQueue() {
        this.stack1 = new Stack<>();//主栈
        this.stack2 = new Stack<>();//辅助栈
    }
    /** 添加元素到队尾 */
    public void add(int x){
        stack1.push(x);

    }

    /** 删除队头的元素并返回 */
    public int deleteHead(){
        // 先调用 peek 保证 s2 非空
        getHead();
        return stack2.pop();
    }

    /** 返回队头元素 */
    public int getHead(){
        //如果s2不为空，返回s2栈顶元素。如果为空将s1的元素都加入s1中，返回s2栈顶元素
        if(stack2.isEmpty()){
            while (!stack1.isEmpty())
                stack2.push(stack1.pop());
        }
        return stack2.peek();
    }

    /** 判断队列是否为空 */
    public boolean empty(){
        return stack1.isEmpty()&&stack2.isEmpty();
    }

    public static void main(String[] args) {
        MyQueue myQueue=new MyQueue();
        myQueue.add(1);
        myQueue.add(2);
        myQueue.add(3);
        myQueue.add(4);
        System.out.println(myQueue.getHead());
        myQueue.deleteHead();
        System.out.println(myQueue.getHead());

    }
}

__EOF__

本文作者：程序员小宇
本文链接：https://www.cnblogs.com/treasury/p/12752410.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！