225. 用队列实现栈225. 用队列实现栈225. 用队列实现栈
请你仅使用两个队列实现一个后入先出(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
都保证栈不为空 -
使用两个队列实现栈,可以使用一个队列作为主队列,另一个队列作为辅助队列,实现如下:
1. 将元素压入栈顶时,将元素插入到主队列的队尾。
2. 移除并返回栈顶元素时,先将主队列中除队尾元素外的所有元素依次出队并插入到辅助队列中,然后将主队列队尾元素出队并返回,最后将辅助队列中的所有元素移动回主队列中。
3. 返回栈顶元素时,先执行与第2步相同的操作,保留主队列队尾元素,然后将辅助队列中的所有元素移动回主队列中,并返回主队列队尾元素。
4. 判断栈是否为空时,只需要判断主队列是否为空即可。
实现 MyStack 类的代码如下:
```c++
class MyStack {
public:
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
q1.push(x);
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
int size = q1.size();
for (int i = 0; i < size - 1; i++) {
q2.push(q1.front());
q1.pop();
}
int top = q1.front();
q1.pop();
for (int i = 0; i < size - 1; i++) {
q1.push(q2.front());
q2.pop();
}
return top;
}
/** Get the top element. */
int top() {
int size = q1.size();
for (int i = 0; i < size - 1; i++) {
q2.push(q1.front());
q1.pop();
}
int top = q1.front();
q2.push(top);
q1.pop();
for (int i = 0; i < size - 1; i++) {
q1.push(q2.front());
q2.pop();
}
return top;
}
/** Returns whether the stack is empty. */
bool empty() {
return q1.empty();
}private:
queue<int> q1;
queue<int> q2;
};
```在这个实现中,q1作为主队列,q2作为辅助队列。push操作直接将元素插入到q1的队尾,pop和top操作需要将q1中的元素移动到q2中,保留q1的队尾元素,然后再将q2中的元素移动回q1中。empty操作只需要判断q1是否为空即可。