剑指 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();
    }
}
原文地址:https://www.cnblogs.com/Vincent-yuan/p/14942647.html