题目
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。
进阶:
你能否实现每个操作均摊时间复杂度为 O(1) 的队列?换句话说,执行 n 个操作的总时间复杂度为 O(n) ,即使其中一个操作可能花费较长时间。
参考别人思路,使用双栈思想
push数据:把数据放进输入栈
pop数据:
输出栈如果为空,就把输入栈数据全部导入,再从输出栈弹出数据
如果输出栈不为空,则直接从输出栈弹出数据就可以了。
如何判断队列为空:如果输入栈和输出栈都为空,则说明模拟的队列为空。
代码
class MyQueue {
public:stack<int> stIn;stack<int> stOut;/** Initialize your data structure here. */MyQueue() {}/** Push element x to the back of queue. */void push(int x) {stIn.push(x);}/** Removes the element from in front of queue and returns that element. */int pop() {//如果输出栈为空if(stOut.empty()){//就把数据全部导入while(!stIn.empty()){stOut.push(stIn.top());stIn.pop();}}//从输出栈弹出数据int result =stOut.top();stOut.pop();return result;}/** Get the front element. */int peek() {//直接使用已有pop函数int res = this->pop();//因为pop()函数弹出了元素res,所以还得再添加回去stOut.push(res);return res;}/** Returns whether the queue is empty. */bool empty() {return stIn.empty() && stOut.empty();}
};/*** Your MyQueue object will be instantiated and called as such:* MyQueue* obj = new MyQueue();* obj->push(x);* int param_2 = obj->pop();* int param_3 = obj->peek();* bool param_4 = obj->empty();*/