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
}


原文地址:https://www.cnblogs.com/davis12/p/15527163.html