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

剑指 Offer 09. 用两个栈实现队列_第1张图片
思路:栈是先进后出,队列是先进先出,故用两个栈来模拟队列。
剑指 Offer 09. 用两个栈实现队列_第2张图片
如图,要添加元素,直接在A中添加即可,但是要删除栈底元素(即模拟队列),直接使用栈A无法删除,故使用辅助栈B将A中的元素逆序存放,然后弹出B的栈顶即可。

//A 用于加入队尾操作,栈 B 用于将元素倒序
class CQueue {
     
    LinkedList<Integer> A, B;
    public CQueue() {
     
        A = new LinkedList<Integer>();
        B = new LinkedList<Integer>();
    }
    public void appendTail(int value) {
     
        //栈A负责队列入队
        A.add(value);
    }
    public int deleteHead() {
     
        //如果B空了,就从A添加
        if (B.isEmpty()){
     
            while(!A.isEmpty()){
     
                B.push(A.pop());
            }
        }
        // B不为空,直接返回B的栈顶元素,如果B为空,说明A中无元素,返回-1
        if(!B.isEmpty()){
     
            return B.pop();
        }else {
     
            return -1;
        }
    }
}

你可能感兴趣的:(剑指offer,队列,栈,用两个栈实现队列)