请你仅使用两个队列实现一个后入先出(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
}