• 小知识，大挑战！本文正在参与“程序员必备小知识”创作活动。

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

• 用两个栈实现一个队列。队列的声明如下，
  • 请实现它的两个函数 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

思路

• 首先分析题目，两个栈去实现一个队列
  • 两个栈如果设置一个进，一个出那么就实现了队列的思想
  • 根据上面思路，可以想到，第一个栈先进
  • 如果第二个栈不为空的话，那么就让第二个栈弹出元素
  • 如果第二个栈为空，那么就让第一个栈弹出元素给第二栈入栈
  • 这样一系列的操作就实现了先进先出的动作

class CQueue {
        LinkedList<Integer> stack1;   //先声明出两个对象
        LinkedList<Integer> stack2;

    public CQueue() {
        // 为两个对象分配空间
        stack1 = new LinkedList<>();
        stack2 = new LinkedList<>();
        
    }
    
    public void appendTail(int value) {
        stack1.push(value);          //append 推进栈1 一个元素
    }
    
    public int deleteHead() {
        if(stack2.isEmpty()){   //栈2 要是为空的话   紧接着判断栈1
            if(stack1.isEmpty()) return -1;

            while(!stack1.isEmpty()){     //要是栈1不为空的话 那就将栈1的元素全都移入到栈2
                stack2.push(stack1.pop());
            }
            return (int)stack2.pop();
        }else{     //要是栈2不为空的话  直接弹出栈2的元素
            return (int)stack2.pop();
        }
        
    }
}