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

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

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

思路:

  1. 栈的特点是先进后出、后进先出,而队列的特点是先进先出、后进后出,所以用一个栈是没办法完成队列的操作的,但是两个栈可以,一个栈用了存储元素,另一个栈用了辅助操作,将第一个栈中的元素依次弹出再压入第二个栈,完成第一个栈元素的清空,第一个栈为空之后,可以将新的元素压入栈底,从而实现了后进后出,第二个栈的元素也可以依次弹出再压入第一个栈,完成元素的还原,如果每个元素过来都这样操作,相当于第一个栈中的元素实现了先来的在栈顶而后来的在栈底,即先进先出和后进后出。
class CQueue {
    Stack<Integer> stack1 = new Stack();
    Stack<Integer> stack2 = new Stack();
    int size;

    public CQueue() {
        stack1 = new Stack();
        stack2 = new Stack();
        size = 0;
    }
    
    public void appendTail(int value) {
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
        stack1.push(value);
        size++;
        while(!stack2.isEmpty()){
            stack1.push(stack2.pop());
        }
    }
    
    public int deleteHead() {
        if(stack1.isEmpty()){
            return -1;
        }else{
            size--;
            return stack1.pop();
        }
    }
}

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */
  1. 上面一种方法是在元素进栈的时候修改顺序,使元素倒序入栈,形成队列,那么也可以在进栈时不修改元素,在删除元素时借助辅助队列完成倒序删除操作。
class CQueue {
    private Deque<Integer> stack1 = null;
    private Deque<Integer> stack2 = null;

    public CQueue() {
        stack1 = new ArrayDeque<>();
        stack2 = new ArrayDeque<>();
    }
    
    public void appendTail(int value) {
        stack1.addLast(value);
    }
    
    public int deleteHead() {
        if (stack2.size() != 0) {
            return stack2.removeLast();
        } 
        if (stack1.size() == 0) {
            return -1;
        } else {
            while (stack1.size() != 0) {
                stack2.addLast(stack1.removeLast());
            }
            return stack2.removeLast();
        }
    }
}

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue obj = new CQueue();
 * obj.appendTail(value);
 * int param_2 = obj.deleteHead();
 */
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值