剑指 Offer 09. 用两个栈实现队列 | 刷题打卡

1. 题目

• 剑指 Offer 09. 用两个栈实现队列
• 难度：简单

2. 描述

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

示例 1：

输入：

["CQueue","appendTail","deleteHead","deleteHead"]

[[],[3],[],[]]

输出：[null,null,3,-1]

示例 2：

输入：

["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]

[[],[],[5],[2],[],[]]

输出：[null,-1,null,null,5,2]

提示：

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

3. 实现方法

3.1 方法 1

3.1.1 思路

1. 队列是一种先进先出的数据结构，而栈是一种先进后出的数据结构，所以要用栈来实现队列，则需要栈，一个用于入队，一个用于出队；
2. 入队时，只需要在对应入队的栈中插入数据即可；
3. 出队时，优先从出队的栈中弹出数据，当出队的栈为空时，再来判断入队的栈是否为空，如果入队的栈为空，则返回 -1，当入队的栈不为空时，将入队的栈中的元素弹出并压入出队的栈中；
4. 最后返回出队的栈中的元素即可；

3.1.2 实现

class CQueue {
    // 两个栈，一个用于入，一个用于出
    // 入的栈 stackIn
    private Stack<Integer> stackIn;
    // 出的栈 stackOut
    private Stack<Integer> stackOut;

    // 构造函数
    public CQueue() {
        stackIn = new Stack<Integer>();
        stackOut = new Stack<Integer>();
    }

    // 插入元素，即在入的栈中压入该元素
    public void appendTail(int value) {
        stackIn.push(value);
    }

    public int deleteHead() {
        // 如果出的栈不为空，则直接从这里边出栈即可
        if(!stackOut.isEmpty()){
            return stackOut.pop();
        }

        // 若出和入的栈均为空，则返回 -1
        if(stackIn.isEmpty()){
            return - 1;
        }

        // 入的栈不为空，则弹出并压入出的栈
        while(!stackIn.isEmpty()){
            stackOut.push(stackIn.pop());
        }

        // 弹出由入的栈弹出并压入出的栈中的元素
        return stackOut.pop();
    }
}

本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情