一、运行结果
二、题目
请你仅使用两个队列实现一个后入先出(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(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-stack-using-queues
三、思路
设置两个队列qu1 和qu2,在qu1中实现栈的操作,qu2辅助qu1实现,在插入元素时,若qu1中有元素,则先将qu1中的全部元素移到qu2中,然后再将新元素插入到qu1,然后再将原来的元素从qu2移回qu1,这样新插入的元素就位于qu1的队头,实现了栈后入先出的特性,而pop(), top() 和 empty()操作都是直接对qu1进行,因为除了插入元素时暂时使用qu2,其他时间qu2都是空的。
四、代码
class MyStack {
public:
queue<int> qu1;
queue<int> qu2;
MyStack() {
}
//在qu1中实现栈的功能,qu2作为辅助
void push(int x) {
while(!qu1.empty()){ //先将qu1中原有的元素全移到qu2
int tmp = qu1.front();
qu1.pop();
qu2.push(tmp);
}
qu1.push(x); //在qu1中插入x
while(!qu2.empty()){
int tmp = qu2.front(); //将qu2中的元素按序移回qu1
qu2.pop();
qu1.push(tmp);
}
}
int pop() {
int tmp = qu1.front();
qu1.pop();
return tmp;
}
int top() { //非插入时,所有的元素都在qu1中
return qu1.front();
}
bool empty() {
return qu1.empty();
}
};