剑指offer 09 两个栈实现队列 解题思路与代码

 

剑指 Offer 09. 用两个栈实现队列

难度简单342收藏分享切换为英文接收动态反馈

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例 1：

输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出：[null,null,3,-1]

示例 2：

输入：
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]

        思路：做这一类型的题可以在纸上先画一下，向栈中加入一些元素，想一下栈和队列两个结构有什么区别？

         *比如加入1,2,3 三个数字，此时想取出，栈弹出的是顶部元素，即为3，但是我们想取出1。这时可以怎么利用第二个栈呢？

        *不难想到可以把第一个栈中的元素依次加入第二个栈，再进行弹出的时候就是遵循我们想要的效果了。

        *这时再完善思路：之后再加入的应该加入到第一个栈当中，取出时先判断第二个栈是不是空的，如果不空，就弹出一个元素。如果是空的，判断第一个栈是不是空的，如果不是空的，就把第一个栈中的元素依次弹出，加入到第二个栈当中。如果第一个栈也是空的，那就说明真的为空。根据题意返回-1就行了.

class CQueue {

    Stack<Integer> s1 ;

    Stack<Integer> s2 ;

    public CQueue() {

        s2 = new Stack<>();

        s1 = new Stack<>();

    }

    

    public void appendTail(int value) {

        s1.push(value);

    }

    

    public int deleteHead() {

        if(s2.empty()){

            if(s1.empty()){

                return -1;

            }

            while(!s1.empty()){

                s2.push(s1.pop());

            }

        }

        return s2.pop();



    }

}



/**

 * Your CQueue object will be instantiated and called as such:

 * CQueue obj = new CQueue();

 * obj.appendTail(value);

 * int param_2 = obj.deleteHead();

 */