算法day28 用栈实现队列

题目描述

思路：双栈
我们知道，队列的规则是先进先出，栈的规则是先进后出，要用栈实现队列的相关操作，我们只需要建立两个栈，一个栈用作输入，一个栈用作输出。当输入时将输入的数据存储在输入栈中，当输出或要查看队列的队首时，我们更新输出栈，然后使用输出栈进行相关的操作。

代码如下

class MyQueue {
public:
stack stack_in;
stack stack_out;
MyQueue() {

}

void push(int x) {//入队操作
stack_in.push(x);
}

int pop() {//出队操作
if(stack_in.empty()&&stack_out.empty()){
return -1;
}
if(stack_out.empty()){//持续更新输出栈
while(!stack_in.empty()){
stack_out.push(stack_in.top());
stack_in.pop();
}
}
int x = stack_out.top();//返回队头元素
stack_out.pop();//出栈
return x;
}

int peek() {
int res = this -> pop();//调用上面完成的出队函数
stack_out.push(res);
return res;
}

bool empty() {
if(stack_in.empty()&&stack_out.empty()){
return true;
}
return false;
}
};

时间复杂度：O(1)
空间复杂度：O(n)

END