题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
题目思路
这题题目考察的是栈和队列,首先这两个有两个非常鲜明的特定,栈:先进后出,队列:先进先出。那么我们就要围绕这两个特点进行对该题的求解。
关于将元素放入(push):其实关于放入,很简单,我们可以就对一个栈进行操作,直接放入栈stack1中即可,stack1.push(元素)。
关于将元素取出也就是删除(pop):从题目可以知道,我们是拥有两个栈的,那么对于取出元素我们可以很容易的想到我们当时先插入元素都已经在一个栈stack1中,此时元素按先进的排后面的顺序排序(符合栈先进后出的特点),然后我们依次取出栈stack1中的元素压入栈stack2中,那么当取完的时候我们便可以在栈stack2中获得到先进先出的元素“队列"。当然要注意的一点是,当我们将栈stack1中元素放入栈stack2中时,我们使用的时栈的top()取栈顶值,pop()删除栈顶元素,这样确保元素都到stack2中时栈stack1已经被清空。这样我们可以直接用stack2.top()取出栈顶元素,stack2.pop()删除栈顶元素,即可完成我们的先进先出。另外取出并删除元素之后我们要用类似的操作将栈stack2中的元素依次重新放回栈stack1中,便于下次放入元素的时候能够放在stack1中的栈顶,符合队列后进后出的特点。
题目代码
class Solution
{
public:
void push(int node) {
stack1.push(node); //进行元素的压入
}
int pop() {
while(!stack1.empty()) //将stack1中的元素压入stack2,
//在stack2中先进的元素更靠近栈顶
{
stack2.push(stack1.top());
stack1.pop();
}
int temp=stack2.top(); //获得stack2中的栈顶元素
stack2.pop();
while(!stack2.empty()) //将stack2进行清空,并且重新压入stack1中
//目的使为了让新进入的元素能在stack1的栈顶
{
stack1.push(stack2.top());
stack2.pop();
}
return temp;
}
private:
stack<int> stack1;
stack<int> stack2;
};