面试题09. 用两个栈实现队列

题目

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

示例 输入：[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”] [[],[3],[],[]]
输出：[null,null,3,-1]

算法思路

栈的特性是先进后出，队列是先进先出，题目要求通过两个栈来实现队列，说明必有一个是辅助栈（可以看做队列），另外一个栈用于存储数据。根据栈的特性，第一个栈底部元素是最先入栈的元素，同时也是第二个栈最先删除的元素。每次向第一个栈插入元素时，如果第二个栈不为空，需要先将第二个栈的元素依次出栈然后重新入第一个栈，然后再将需要入栈的元素value入栈；需要删除元素时，删除的是刚刚入第一个栈的元素，在第一个栈的栈头，次时需要将第一个栈的元素依次删除如第二个辅助栈，第二个辅助栈的头元素即为第一个栈的栈尾，相当于辅助栈是第一个栈反转存储的。然后将辅助栈删除pop()即实现了先进先出。

代码

class CQueue {
    Stack<Integer> stack1;
    Stack<Integer> stack2 ;
    public CQueue() {
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    }
    
    public void appendTail(int value) {    
        while(!stack2.isEmpty()){  //先判断栈2是否为空，value一定是在栈头
            stack1.push(stack2.pop());
        }           
        stack1.push(value);
    }
    
    public int deleteHead() {
        while(!stack1.isEmpty()) {  //注意是while，不是if
            stack2.push(stack1.pop());
        }     
        if(!stack2.isEmpty()){
            return stack2.pop();
        }           
        return -1;  
    }
}
/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */