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

用两个栈实现一个队列。

队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，

分别完成在队列尾部插入整数和在队列头部删除整数的功能。

(若队列中没有元素，deleteHead 操作返回 -1 )

示例 1：

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

提示：

• 1 <= values <= 10000
• 最多会对 appendTail、deleteHead 进行 10000 次调用

 

题解：

使用两个栈stack1,stack2,其中stack1负责插入，stack2负责删除。

删除时，如果stack2为空，则先把stack1的数据出到stack2中，

然后再stack2出栈，如果此时stack2仍为空，则返回-1.

public class offer9 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub

        CQueue obj = new CQueue();
        obj.appendTail(3);
        
        int param = obj.deleteHead();
        int param_1 = obj.deleteHead();
        System.out.println(param_1);
    }

}

class CQueue{
    private Stack stack1; 
    private Stack stack2;
    
    public CQueue(){
        stack1 = new Stack();
        stack2 = new Stack();
    }
    
    public void appendTail(int value){
        stack1.add(value);
    }
    
    public int deleteHead(){
        if(stack2.isEmpty()){
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.isEmpty()?-1:(int) stack2.pop();
    }
}