题目:请你仅使用两个队列实现一个后入先出(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
思路:用两个队列实现栈。定义两个队列,queue1和queue2。
入栈时将数据放入queue1。
出栈时,先将queue1中除最后一个元素之后的其余元素都移入queue2,然后将queue1中的最后一个元素弹出。再讲queue2中的所有元素移回queue1。
返回栈顶元素可以调用已经实现好的pop(),将弹出的值保存,再调用push()压入。
判断栈是否为空即判断queue1是否为空。
代码:
class MyStack { //用队列实现栈
queue<int> queue1;
queue<int> queue2;
public:
MyStack() {
}
void push(int x) {
queue1.push(x);
}
int pop() {
int len = queue1.size() - 1;
while (len--) { //将queue1中除了最后一个元素之外的元素移到queue2
queue2.push(queue1.front());
queue1.pop();
}
int res = queue1.front();
queue1.pop();
while (!queue2.empty()) { //将queue2中的所有元素移回queue1
queue1.push(queue2.front());
queue2.pop();
}
return res;
}
int top() {
int res = this->pop();
this->push(res);
return res;
}
bool empty() {
return queue1.empty();
}
};