剑指Offer第七题——两个栈实现一个队列以及两个队列实现一个栈

那么多学技术的都可以成功，凭什么我不行

两个栈实现一个队列以及两个队列实现一个栈

更多《剑指Offer》Java实现合集

 

目录

两个栈实现队列

题目

 思路

代码实现

 

收获

延申 两个队列实现一个栈

思路

代码实现

 

 

 

---

两个栈实现队列

题目

　 用两个栈实现一个队列。请实现它的两个函数appendTail和deleteHead，
 分别完成在队列尾部插入结点和在队列头部删除结点的功能。

 思路

　　这道题较简单，自己先试着模拟一下插入删除的过程（在草稿纸上动手画一下）：插入肯定是往一个栈stack1中一直插入； 
删除时，直接出栈无法实现队列的先进先出规则，这时需要将元素从stack1出栈，压到另一个栈stack2中，然后再从stack2中出栈就OK了。  需要稍微注意的是：当stack2中还有元素，stack1中的元素不能压进来； 
当stack2中没元素时，stack1中的所有元素都必须压入stack2中。否则顺序就会被打乱。

代码实现

 

class Queue {
    private Stack<Integer> stack1 = new Stack<>();
    private Stack<Integer> stack2 = new Stack<>();

    public void appendTail(Integer node) {
        stack1.push(node);
    }

    public int deleteHead() {
        //如果stack2为空
        if (stack2.isEmpty()) {
            if (stack1.isEmpty()) {//如果stack1为空
                throw new RuntimeException("队列为空");
            } else {//stack1不为空
                while (!stack1.isEmpty()) {//将stack1中的元素全部压入stack2中
                    stack2.push(stack1.pop());
                }
            }
        }
        return stack2.pop();
    }
}

收获

用画图将抽象问题形象化

延申 两个队列实现一个栈

思路

由于栈的性质是“后进先出”的，用两个队列模拟实现栈的时候就需要两个队列的元素“互捣”，从而实现栈的这一特性

具体的做法，我们还是用一张图来看一看

代码实现

class MyStack {
    private Queue<Integer> queue1 = new LinkedList<>();//Queue是接口，不能直接实例化
    private Queue<Integer> queue2 = new LinkedList<>();

    /*入栈*/
    public void push(int item) {
        //如果两个队列都为空，优先选择queue1来添加元素
        if (queue1.isEmpty() && queue2.isEmpty()) {
            queue1.add(item);
            return;
        }
        //如果queue1为空，向queue2添加元素
        if (queue1.isEmpty()) {
            queue2.add(item);
        } else {
            queue1.add(item);
        }
    }

    /*出栈*/
    public int pop() {
        //如果两个队列都为空
        if (queue1.isEmpty() && queue2.isEmpty()) {
            throw new RuntimeException("栈为空");
        }

        //如果queue2为空，将queue1的前queue1.size()-1个元素弹出放入queue2队列，queue1的最后一个元素直接弹出
        if (queue2.isEmpty()) {
            while (queue1.size() > 1) {
                queue2.add(queue1.poll());
            }
            return queue1.poll();
        }
        //如果queue1为空，将queue2的前queue2.size()-1个元素弹出放入queue1队列，queue2的最后一个元素直接弹出
        if (queue1.isEmpty()) {
            while (queue2.size() > 1) {
                queue1.add(queue2.poll());
            }
            return queue2.poll();
        }
        return 0;
    }
}

更多《剑指Offer》Java实现合集