请你仅使用两个队列实现一个后入先出(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
(注意:队列是先进先出,栈是后进先出)
双队列实现 解题思路:
建立两个队列,一个主队列用来存储栈,一个辅队列用来辅助实现出栈操作。
入栈操作:直接存入主队列
返回栈顶元素:直接返回主队列的队尾元素,可以用back()函数返回队尾元素。
出栈操作:对应的是把主队列的队尾元素出队,这时可以先把主队列除队尾元素全部依次出队并依次入队辅队列,返回该队尾元素并让其出队,最后再把辅队列赋给主队列。
判断栈是否为空:可以直接判断主队列是否为空,因为主队列是用来存储栈的。
代码实现:
// 无注释版
class MyStack {
public:
queue<int>que1;
queue<int>que2;
MyStack() {
}
void push(int x) {
que1.push(x);
}
int pop() {
int i=que1.size()-1;
while(i--) {
que2.push(que1.front());
que1.pop();
}
int result=que1.front();
que1.pop();
que1=que2;
while(!que2.empty()) {
que2.pop();
}
return result;
}
int top() {
return que1.back();
}
bool empty() {
return que1.empty();
}
};
// 注释版
class MyStack { // 类实现
public:
queue<int>que1; // 创建主队列que1
queue<int>que2; // 创建辅队列que2
MyStack() {
}
// 进栈操作:直接入队主队列
void push(int x) {
que1.push(x);
}
// 出栈操作
int pop() {
// 让主队列除队尾元素全部出队并入队辅队列
int i=que1.size()-1;
while(i--) {
que2.push(que1.front());
que1.pop();
}
int result=que1.front(); // 记下队尾元素
que1.pop(); // 清空主队列
que1=que2; // 把辅队列赋给主队列
// 清空辅队列
while(!que2.empty()) {
que2.pop();
}
return result; // 返回队尾元素(出栈)
}
// 返回栈顶元素:直接返回主队列队尾元素
int top() {
return que1.back();
}
// 判断栈是否为空:直接判断主队列是否为空
bool empty() {
return que1.empty();
}
};
单队列实现 解题思路:
除了出栈操作其他都一样,可以参考双队列实现,在此略过。
出栈操作:直接把队头元素出队再入队,循环直到原始队尾元素为止。再把对尾元素出队并返回。完成出栈操作。
代码实现:
class MyStack {
public:
queue<int>que;
MyStack() {
}
void push(int x) {
que.push(x);
}
int pop() {
int i=que.size()-1;
while(i--){
que.push(que.front()); //队头元素入队
que.pop(); // 队头元素出队
}
int result=que.front(); //记下元素队尾元素
que.pop(); // 原始队尾元素出队
return result; // 返回原始队尾元素
}
int top() {
return que.back();
}
bool empty() {
return que.empty();
}
};