题目:用两个栈实现队列
题目描述:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
题目思考:
首先,我们需要回忆一下两个概念:栈和队列,以及其各自的特性。
栈:先进后出
队列:先进先出
栈的STL相关方法:
入栈,s.push(x)
出栈,s.pop()
访问栈顶,s.top()
判断栈空,s.empty()
访问栈中的元素个数,s.size()
队列的STL相关方法
入队,q.push(x)
出队,q.pop()
访问队首元素,q.front()、访问队尾元素,q.back()
判断队列空,q.empty()
访问队列中的元素个数,q.size()
题目分析:
队列的特性:先进先出,我们针对这个特性用两个栈来实现。
push(入队)的时候,只要stack1中不断的加入就好。
pop(出队)的时候,我们先判断stack2中有没有元素,没有将stack1中的数据出栈,push入栈到stack2中,之后我们只需要将stack2中的元素出栈就实现了队列的先进先出。
题目代码:
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
if(stack2.empty()){
while(!stack1.empty()){
stack2.push(stack1.top());
stack1.pop();
}
}
int result = stack2.top();
stack2.pop();
return result;
}
private:
stack<int> stack1;
stack<int> stack2;
};