咋又要用队列实现栈了嘛?!

简介: 咋又要用队列实现栈了嘛?!

大家好呀,我是蛋蛋。


今天解决用队列实现栈这道题,和用栈实现队列一样是考察对”栈和队列理解程度“的好题。


但是用队列实现栈,和用栈实现队列是两种不同的移动思路。


这种题都是为了考察而考察,没啥工程意义,实际工作中要是有人敢让你这么做,不要说话,先给他一 jue。


话不多说,直接开整。

640.png


   LeetCode 232:用栈实现队列


题意


用队列实现栈的下列操作:


void push(int x):元素 x 入栈。


int pop():移除并返回栈顶元素。


int peek():返回栈顶元素。


boolean empty():栈为空,返回 true,否则返回 false。


示例

640.png

提示


  • 1 <= x <= 9
  • 最多调用 100 次 push、pop、top 和 empty
  • 每次调用 pop 和 top 都保证栈不为空


题目解析


水题,难度简单,主要还是考察对栈和队列的理解能力


如果对栈和队列还不熟悉,看一下本帅比写的这篇文章:


呔!“栈”住,队列!


和栈实现队列相仿,用队列实现栈也涉及 4 步操作:


  • 入栈 push
  • 出栈 pop
  • 取栈顶元素 peek
  • 判空 empty


知道了需求,下面就是如何用队列模拟栈。


这里有两种方法:


  1. 两个队列模拟。
  2. 一个队列模拟。


两个队列模拟


建一个主队列和一个辅助队列,每次入栈操作时,将新元素添加到辅助队列,再依次将主队列的元素出队列,依次加入辅助队列,最后将主队列与辅助队列互换。


一个队列模拟


建一个主队列,其余操作正常,只是在每次模拟栈弹出元素的时候,将除了最后一个元素外的之前元素弹出队列,重新添加到主队列尾部即可。


图解


为了照顾大多数臭宝,我用两个队列来模拟,笨笨的方法更直观一些。


首先初始化主队列和辅助队列。

640.png

def __init__(self):
    self.major_queue = []
    self.help_queue = []

push(x),入栈操作,先压入辅助队列,再将主队列的元素加入辅助队列,再互换主队列和辅助队列。


首先 push(1):

640.png


再 push(2):

640.png


def push(self, x: int) -> None:
    # 新元素放入辅助队列的队首
    self.help_queue.append(x)
    # 将主队列的元素加入辅助队列
    while self.major_queue:
        self.help_queue.append(self.major_queue.pop(0))
    # 现在辅助队列元素是和栈的元素排列一致,为了后续方便理解,交换两个队列的内容
    self.major_queue = self.help_queue
    self.help_queue = []

入栈需要将所有的元素都倒腾一遍,所以入栈的时间复杂度为 O(n),因为需要额外的队列存储栈内的元素,所以入栈的空间复杂度为 O(n)。


pop(),出栈操作,出栈出的是栈顶元素,由上图可以看出,栈顶元素即主队列的队首元素。

640.png


def pop(self) -> int:
    # 判断是否为空
    if self.empty():
        return None
    # pop移除栈顶元素,相当于移除主队列的队首元素
    return self.major_queue.pop(0)


出栈对应的是将主队列的队首元素出栈,所以时间复杂度是 O(1),空间复杂度为 O(n)。


top(),获取栈顶元素,与 pop 类似,即相当于获取主队列的队首元素。

def top(self) -> int:
    # top获取栈顶元素,即获取主队列的队首元素
    return self.major_queue[0]

同样,获取栈顶元素是获取主队列的首元素,所以时间复杂度为 O(1),空间复杂度为 O(n)。


empty(),判断栈是否为空。其实就是看主队列是否为空,主队列为空,栈就空,主队列不为空,栈不为空。

def empty(self) -> bool:
    # 只需要判断主队列是否为空
    if not self.major_queue:
        return True
    return False

对于判空,只需要判断主队列是否为空,所以时间复杂度为 O(1),空间复杂度为 O(n)。


代码实现


Python 代码实现

class MyStack:
    def __init__(self):
        """
        Initialize your data structure here.
        """
        # 初始化主队列和辅助队列
        self.major_queue = []
        self.help_queue = []
    def push(self, x: int) -> None:
        """
        Push element x onto stack.
        """
        # 新元素放入辅助队列的队首
        self.help_queue.append(x)
        # 将主队列的元素加入辅助队列
        while self.major_queue:
            self.help_queue.append(self.major_queue.pop(0))
        # 现在辅助队列元素是和栈的元素排列一致,为了后续方便理解,交换两个队列的内容
        self.major_queue = self.help_queue
        self.help_queue = []
    def pop(self) -> int:
        """
        Removes the element on top of the stack and returns that element.
        """
        # 判断是否为空
        if self.empty():
            return None
        # pop移除栈顶元素,相当于移除主队列的队首元素
        return self.major_queue.pop(0)
    def top(self) -> int:
        """
        Get the top element.
        """
        # top获取栈顶元素,即获取主队列的队首元素
        return self.major_queue[0]
    def empty(self) -> bool:
        """
        Returns whether the stack is empty.
        """
        # 只需要判断主队列是否为空
        if not self.major_queue:
            return True
        return False
# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()


Java 代码实现

class MyStack {
    Queue<Integer> queue1; // 和栈中保持一样元素的队列
    Queue<Integer> queue2; // 辅助队列
    /** Initialize your data structure here. */
    public MyStack() {
        queue1 = new LinkedList<>();
        queue2 = new LinkedList<>();
    }
    /** Push element x onto stack. */
    public void push(int x) {
        queue2.offer(x); // 先放在辅助队列中
        while (!queue1.isEmpty()){
            queue2.offer(queue1.poll());
        }
        Queue<Integer> queueTemp;
        queueTemp = queue1;
        queue1 = queue2;
        queue2 = queueTemp; // 最后交换queue1和queue2,将元素都放到queue1中
    }
    /** Removes the element on top of the stack and returns that element. */
    public int pop() {
        return queue1.poll(); // 因为queue1中的元素和栈中的保持一致,所以这个和下面两个的操作只看queue1即可
    }
    /** Get the top element. */
    public int top() {
        return queue1.peek();
    }
    /** Returns whether the stack is empty. */
    public boolean empty() {
        return queue1.isEmpty();
    }
}


图解队列实现栈到这就结束啦,是不是觉得小菜一碟?


别飘,这只是简单题,后面题解继续。


如果有问题的话,直接评论区见呀,动动小手,给我你的点赞在看么么哒


我是帅蛋,我们下次见!


640.gif

相关文章
|
1天前
|
算法
数据结构与算法:栈与队列
数据结构与算法:栈与队列
|
1天前
|
存储 缓存 算法
堆和栈的区别及应用场景
堆和栈的区别及应用场景
|
4天前
|
存储 算法 调度
算法与数据结构-栈篇
算法与数据结构-栈篇
11 0
|
5天前
|
存储 人工智能 程序员
技术心得记录:堆(heap)与栈(stack)的区别
技术心得记录:堆(heap)与栈(stack)的区别
|
7天前
【海贼王的数据航海】栈和队列
【海贼王的数据航海】栈和队列
6 0
|
7天前
|
算法 C语言
【数据结构与算法 经典例题】使用栈实现队列(图文详解)
【数据结构与算法 经典例题】使用栈实现队列(图文详解)
|
7天前
|
算法 C语言
【数据结构与算法 经典例题】使用队列实现栈(图文详解)
【数据结构与算法 经典例题】使用队列实现栈(图文详解)
|
7天前
|
算法 程序员 数据处理
【数据结构与算法】使用单链表实现队列:原理、步骤与应用
【数据结构与算法】使用单链表实现队列:原理、步骤与应用
|
7天前
|
存储 算法 编译器
【数据结构与算法】使用数组实现栈:原理、步骤与应用
【数据结构与算法】使用数组实现栈:原理、步骤与应用
|
7天前
|
算法
【数据结构和算法】---栈和队列的互相实现
【数据结构和算法】---栈和队列的互相实现
7 0