zoukankan      html  css  js  c++  java
  • 0225-leetcode算法实现之用队列实现栈-implement-stack-using-queues-python&golang实现

    请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。

    实现 MyStack 类:

    void push(int x) 将元素 x 压入栈顶。
    int pop() 移除并返回栈顶元素。
    int top() 返回栈顶元素。
    boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。

    注意:

    你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。
    你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。

    示例:

    输入:
    ["MyStack", "push", "push", "top", "pop", "empty"]
    [[], [1], [2], [], [], []]
    输出:
    [null, null, null, 2, 2, false]

    解释:
    MyStack myStack = new MyStack();
    myStack.push(1);
    myStack.push(2);
    myStack.top(); // 返回 2
    myStack.pop(); // 返回 2
    myStack.empty(); // 返回 False

    提示:

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

    进阶:你能否实现每种操作的均摊时间复杂度为 O(1) 的栈?换句话说,执行 n 个操作的总时间复杂度 O(n) ,尽管其中某个操作可能需要比其他操作更长的时间。你可以使用两个以上的队列。

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/implement-stack-using-queues

    python

    # 0225.队列实现栈
    
    class MyStack:
        """
        一个队列为主队列,一个为辅助队列,当入栈操作时,
        我们先将主队列内容导入辅助队列,然后将入栈元素放入主队列队头位置,
        再将辅助队列内容,依次添加进主队列即可。
        """
        def __init__(self):
            from collections import deque
            self.queue1 = deque() # 主队列
            self.queue2 = deque() # 辅助队列, 仅在入栈时操作
    
        def push(self, x: int):
            # 先添加到q2的右侧尾部
            self.queue2.append(x)
            # q1非空时,q1 pop左侧元素(q1出队)进q2的右侧队尾,直到q1为空
            while self.queue1:
                self.queue2.append(self.queue1.popleft())
            # 交换q1,q2,完成入栈操作
            self.queue1, self.queue2 = self.queue2, self.queue1
    
        def pop(self):
            # q1队头元素出队即可
            return self.queue1.popleft()
    
        def top(self):
            # get 栈顶元素
            return self.queue1[0]
    
        def empty(self):
            return not self.queue1
    
    
    class MyStack1:
        """
        一个队列实现,
        - push,push入栈时,队头元素ele1出队后,待push新元素newEle后再入队ele1
        - pop, 队头元素出队
        - top, get队头元素
        - empty, 队空即空
        """
        def __init__(self):
            from collections import deque
            self.queue = deque()
    
        def push(self, x: int):
            # 新元素入栈,依次把前n个队头元素放在队尾,保持入栈元素在队头
            n = len(self.queue)
            self.queue.append(x)
            for _ in range(n):
                self.queue.append(self.queue.popleft())
    
        def pop(self):
            # 队头元素出队,栈顶元素出栈
            return self.queue.popleft()
    
        def top(self):
            # get 队头元素,获取栈顶元素
            return self.queue[0]
    
        def empty(self):
            # 队空,栈空
            return not self.queue
    
    

    golang

    package main
    
    // 单队列实现
    type MyStack struct {
    	queue []int
    }
    
    func Constructor() (s MyStack) {
    	return
    }
    
    // push
    func (s *MyStack) Push(x int) {
    	n := len(s.queue)
    	s.queue = append(s.queue, x) // 加入队列
    	for ; n > 0; n-- {           //队头元素加入队列尾部,更新队列
    		s.queue = append(s.queue, s.queue[0])
    		s.queue = s.queue[1:]
    	}
    }
    
    // pop
    func (s *MyStack) Pop() int {
    	val := s.queue[0]     // 取队头元素
    	s.queue = s.queue[1:] // 更新队列
    	return val            // 返回元素,队头元素-栈顶元素出栈
    }
    
    // top
    func (s *MyStack) Top() int {
    	return s.queue[0]
    }
    
    // empty
    func (s *MyStack) Empty() bool {
    	return len(s.queue) == 0
    }
    
    // 双队列实现
    type Mystack1 struct {
    	queue1, queue2 []int
    }
    
    func Constructor() (s *Mystack1) {
    	return
    }
    
    // push
    func (s *Mystack1) Push(x int) {
    	s.queue2 = append(s.queue2, x)
    	for len(s.queue1) > 0 {
    		s.queue2 = append(s.queue2, s.queue1[0])
    		s.queue1 = s.queue1[1:]
    	}
    	s.queue1, s.queue2 = s.queue2, s.queue1
    }
    
    // pop
    func (s *Mystack1) Pop() int {
    	val := s.queue1[0]
    	s.queue1 = s.queue1[1:]
    	return val
    }
    
    // top
    func (s *Mystack1) Top() int {
    	return s.queue1[0]
    }
    
    // empty
    func (s *Mystack1) Empty() bool {
    	return len(s.queue1) == 0
    }
    
    
    
  • 相关阅读:
    计蒜客 奇怪的国家
    计蒜客 泥塑课
    计蒜客 判断质数
    hiho #1143 : 骨牌覆盖问题·一 (运用快速幂矩阵)
    二叉树建立,先序、中序、后序遍历(c实现)
    hiho #1272 买零食 [Offer收割]编程练习赛2
    hiho #1283 hiho密码 [Offer收割]编程练习赛3
    hiho #1288 微软2016.4校招笔试题 Font Size
    hiho一下 第九十八周 搜索一·24点
    hiho一下 第九十七周 数论六·模线性方程组
  • 原文地址:https://www.cnblogs.com/davis12/p/15527163.html
Copyright © 2011-2022 走看看