题目描述
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty): 实现 MyQueue 类:
- void push(int x) 将元素 x 推到队列的末尾
- int pop() 从队列的开头移除并返回元素
- int peek() 返回队列开头的元素
- boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明: 你 只能 使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
题解
堆栈的特性是 LIFO,而队列则是 FIFO,因此想要用堆栈实现队列,需要用两个堆栈的 LIFO 叠加效果,来实现 FIFO。
class MyQueue {
public:
MyQueue() = default;
void push(int x) {
while (!output_.empty()) {
input_.push(output_.top());
output_.pop();
}
input_.push(x);
}
int pop() {
while (!input_.empty()) {
output_.push(input_.top());
input_.pop();
}
int ret = output_.top();
output_.pop();
return ret;
}
int peek() {
while (!input_.empty()) {
output_.push(input_.top());
input_.pop();
}
return output_.top();
}
bool empty() {
return input_.empty() && output_.empty();
}
private:
std::stack<int> input_;
std::stack<int> output_;
};
评论