题目描述
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
示例:
输入:
["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
解题思想
用两个队列que1和que2实现队列的功能,que2其实完全就是一个备份的作用,把que1最后面的元素以外的元素都备份到que2,然后弹出最后面的元素,再把其他元素从que2导回que1。
代码
cpp
class MyStack {
public:
queue<int> qu1;
queue<int> qu2;
MyStack() {
}
void push(int x) {
qu1.push(x);
}
int pop() {
//留下qu1队列中的最后一个元素
while (qu1.size() != 1) {
qu2.push(qu1.front());
qu1.pop();
}
int result = qu1.front();
qu1.pop();
//再将qu2的元素全放到qu1中
qu1 = qu2;
while (!qu2.empty()) {
qu2.pop();
}
return result;
}
int top() {
return qu1.back();
}
bool empty() {
return qu1.empty();
}
};
解题思想2
使用一个队列就能实现,pop的时候将最后一个元素之前的元素全部重新入队。
代码
cpp
class MyStack {
public:
queue<int> qu1;
MyStack() {
}
void push(int x) {
qu1.push(x);
}
int pop() {
int size = qu1.size();
while (size!=1) {
qu1.push(qu1.front());
qu1.pop();
size--;
}
int result = qu1.front();
qu1.pop();
return result;
}
int top() {
return qu1.back();
}
bool empty() {
return qu1.empty();
}
};