LeetCode 第232题 用栈实现队列题(Java解法)- 剑指 Offer 09. 用两个栈实现队列

1:问题描述

来源:LeetCode

难度:简单


问题详情:

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTaildeleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,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]

如果不知道输入和输出是什么意思的,可以参考我的另一篇博文中的2.1小节


2:问题分析与解决

2.1 问题分析

使用两个栈来实现队列。队列的特点就是“先入先出”,也就是先来的永远排在最前面,最先出去的也是最先进来的,后来的只能排末尾。

class CQueue {
    
    


    public CQueue() {
    
    

    }

    public void appendTail(int value) {
    
    
    }

    public int deleteHead() {
    
    
    }
}

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

两个栈,一个自然的想法是:当需要将元素插入到队尾时,将栈s1中的元素全部弹出并压入到另一个空栈s2中,在s1中压入新元素x.然后再把s2中的元素再放回s1,这样就实现将元素插入到队列的尾部的功能。

但是这样繁琐的弹出压入操作,会导致较高的时间消耗。

一个更合理的操作时,插入到队尾时,就正常的入栈到s1。此时栈的顺序和所需的队列顺序是相反的。(而添加元素的函数是不需要返回值的,所以不会影响输出)。

当需要删除头元素时,由于头元素此时在s1的栈尾,我们需要把这个栈中的元素弹出再压入到s2中,此时头元素就在s2的栈顶了,然后弹出s2的栈顶元素即可。

2.2 双栈

2.2.2 代码

class CQueue {
    
    

    Stack<Integer> s1, s2;

    public CQueue() {
    
    
        s1 = new Stack<Integer>();
        s2 = new Stack<Integer>();

    }

    public void appendTail(int value) {
    
    
        s1.push(value);
    }

    public int deleteHead() {
    
    
        if (s2.isEmpty()) {
    
    
            if (s1.isEmpty())
                return -1;
            while (!s1.isEmpty()) {
    
    
                s2.push(s1.pop());
            }
        }
        return s2.pop();
    }
}

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

猜你喜欢

转载自blog.csdn.net/weixin_43490422/article/details/126760628