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

队列的特点是先进先出,而栈的特点是先进后出,但是这道题我们可以利用两个栈来实现队列的特点。

定义了两个栈,queue作为主要的栈,aux为辅助队列。初始时刻优先将数值放入到queue中,当queue中有一个数值时,就将余下的数值放入到aux中。队列首个数值出队时,这是queue中是空,然后需要将aux中的数值弹出并压入到queue中,这时的顺序已经翻转过来了,再下次出队时,直接pop即可。当queue为空时,执行前述步骤。

代码如下:

class CQueue {
    Stack<Integer> queue;
    Stack<Integer> aux;
    public CQueue() {
        queue = new Stack<>();
        aux = new Stack<>();
    }
    
    public void appendTail(int value) {
        if (queue.isEmpty()) queue.push(value);
        else aux.push(value);
    }
    
    public int deleteHead() {
        if (queue.isEmpty()) return -1;
        int ans = queue.pop();
        if (queue.isEmpty()) {
            while (!aux.isEmpty()) queue.push(aux.pop());
        }
        return ans;
    }
}

路飞题解:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/solution/mian-shi-ti-09-yong-liang-ge-zhan-shi-xian-dui-l-2/ 

 

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页