[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.用两个队列实现栈
思路:一个work队列用于添加元素,tmp队列用于临时存放元元素,出栈时,将除了队尾元素之外的所有元素依次出队并放入tmp中,然后在出队,该元素就是栈顶元素val,记录该元素,并将tmp的所有元素放回work中。
2.进阶:用一个队列实现栈
思路:和1一样,只不过将除了尾部元素之外的所有元素依次出队并再次入队,完毕之后,队列头部就是栈顶元素!
class MyStack {
queue<int> work; //工作队列
queue<int> tmp; //辅助队列
public:
MyStack() {
}
void push(int x) {
work.push(x);
}
int pop() {
//将work队列中,除开队尾之外的所有元素放置在tmp中,然后出队,最终在将tmp的所有元素放回work中
int size = work.size();
while(size > 1){
int val = work.front();
tmp.push(val);
work.pop();
}
int val = work.front();
work.pop();
size = tmp.size();
while(size > 0){
int val = tmp.front();
work.push(val);
tmp.pop();
}
return val;
}
int top() {
if(!empty()){
int val = pop();
work.push(val);
return val;
}
return -1;
}
bool empty() {
return work.empty();
}
};
class MyStack {
queue<int> work; //工作队列
public:
MyStack() {
}
void push(int x) {
work.push(x);
}
int pop() {
//将work队列中,除开队尾之外的所有元素放置在tmp中,然后出队,最终在将tmp的所有元素放回work中
int size = work.size();
while(size > 1){
int val = work.front();
work.push(val);
work.pop();
size--;
}
//此时对头元素就是栈的栈顶元素
int val = work.front();
work.pop();
return val;
}
int top() {
if(!empty()){
int val = pop();
work.push(val);
return val;
}
return -1;
}
bool empty() {
return work.empty();
}
};