剑指offer:2个栈实现队列

Posted 2021-01-31 fabaogege

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

思路：栈的特点后进先出，队列的特点是先进先出。2个栈的情况可以通过2次的压栈和出栈实现队列的先进先出。

其中一个栈来实现入队，一个栈实现出队。

图示：

注意：对于入队操作，直接对stack1进行入栈操作就好了。

对于出队操作，需要有几个条件判断一下：

1.当stack1和stack2都为空的时候，出队操作是不合理的，必须进行判断

2.由于出队的操作是在stack2中进行的。首先stack2为空的时候，就需要先把stack1的元素弹出并压入到stack2中，stack2再弹出，这样就完成了出队操作。

当stack2不为空的时候，由于stack2的数据都是stack2出栈来的，顺序是一定的，所以直接弹出stack2栈顶元素就好。

import java.util.Stack;

public class StackToQueue {
　　Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
     //入队：即为stack1的入栈
    public void push(int node) {
        stack1.push(node);
    }
     //出队
    public int pop() {
        //队列为空判断
        if(stack1.empty()&&stack2.empty()){
            throw new RuntimeException("队列为空");
        }
        //分stack2是否为空进行判断。
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

　github源码：栈实现队列操作

图片来源：https://cuijiahua.com/blog/2017/11/basis_5.html