题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（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。

 1class MyQueue {
 2public:
 3    MyQueue() = default;
 4
 5    void push(int x) {
 6        while (!output_.empty()) {
 7            input_.push(output_.top());
 8            output_.pop();
 9        }
10        input_.push(x);
11    }
12
13    int pop() {
14        while (!input_.empty()) {
15            output_.push(input_.top());
16            input_.pop();
17        }
18        int ret = output_.top();
19        output_.pop();
20        return ret;
21    }
22
23    int peek() {
24        while (!input_.empty()) {
25            output_.push(input_.top());
26            input_.pop();
27        }
28        return output_.top();
29    }
30
31    bool empty() {
32        return input_.empty() && output_.empty();
33    }
34private:
35    std::stack<int> input_;
36    std::stack<int> output_;
37};