力扣大战——用两个栈实现队列(对应剑指 Offer 09)

DBC 1.3K 0

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

输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]
温馨提示

一般来说题目可能看花眼,这里引用评论的解释。

简单明了,带你直接看懂题目和例子。 输入: ["CQueue","appendTail","deleteHead","deleteHead"] 这里是要执行的方法,从左到右执行

[[],[3],[],[]]对应上面的方法,是上面方法的参数。CQueue和deleteHead方法不需要指定数字,只有添加才需要指定数字

1.创建队列,返回值为null

2.将3压入栈,返回值为null

3.将栈底的元素删除,也就是消息队列中先进来的元素,所以是deleteHead,返回该元素的数值,所以为3

4.继续删除栈底的元素,但是没有元素了,所以返回-1

所以就有了下面的输出 输出:[null,null,3,-1]

示例 2: 输入: ["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]

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

1.创建队列,返回值为null

2.删除栈底的元素,但是没有元素,所以返回-1

3.把5压入栈,返回null

4.把2压入栈,返回null

5.删除栈底的一个元素,也就是消息队列中先进来的元素,所以是deleteHead,就是最先进来的5,返回值为5,

6.删除栈底的一个元素,就是后进来的2,返回值为2,

所以就有了下面的输出

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

有没有发现先进来的数字,首先显示出来了,但是题目中说要使用栈,栈是先进后出的,使用栈来实现先进先出,在这里使用两个栈就好了,从一个进来再到另一个栈,这样顺序就是先进先出了。题目的主旨写在第一句,就是,使用两个栈实现一个队列。

先上代码

import java.util.LinkedList;
import java.util.Stack;

/**
 * @author DBC
 * @date 2022/5/26 10:23
 * @network dbc655.top
 */
class CQueue {
    LinkedList<Integer> A, B;
    public CQueue() {
        A = new LinkedList<Integer>();
        B = new LinkedList<Integer>();
    }
    public void appendTail(int value) {
        A.addLast(value);
    }
    public int deleteHead() {
        if(!B.isEmpty()) {
            return B.removeLast();
        }
        if(A.isEmpty()) {
            return -1;
        }
        while(!A.isEmpty()) {
            B.addLast(A.removeLast());
        }
        return B.removeLast();
    }
}

大白话解析

appendTail
首先,我们需要在队尾添加一个元素,我们知道栈是一个数据一个数据压入栈中,后面进来的一定在后面,那么我们几乎不需要动脑,直接压入栈中就好了!
deleteHead
我们知道,栈是先进后出的,所以我们需要一个个从最后面取出来,取到最后一个才是我们想要的数据,那我们如何删除栈顶的数据呢,我们可以将A栈的数据全部压入B栈中,正因为先进后出,我们压入B栈后,B栈的最后一个元素就是A栈的第一个元素,那么我们删除B栈的最后一个元素即可!

发表评论 取消回复
表情 图片 链接 代码

分享