题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
(1)插入:我们知道栈是一个“先进后出”的数据结构,而队列是一个“先进先出”的数据结构。现在要求用两个“先进后出”的栈来实现一个“先进先出”的队列。现在我们通过一个具体实例来分析如何实现。首先插入一个元素a,不妨先把它插入到stack1中,此时stack1中只有{a}。接着在插入两个元素b和c,同样插入到stack1中,此时stack1中有{a,b,c},其中c是位于栈顶的。而stack2为空。
(2)删除:现在我们要删除一个元素,按照队列“先进先出”的特性,此时应该删除元素a。此时a存储在stack1中但并不在栈顶,不能直接删除。那怎么办呢,这时候就该stack2上场了,我们把stack1中的元素逐个弹出并压入stack2中,经过3次弹出stack1和压入stack2,stack1为空,stack2元素为{c,b,a},此时stack2中的顺序正好和stack1中的顺序相反,a位于栈顶,可以直接弹出。继续删除元素,接下来应该删除b,此时b位于stack2的栈顶,可以直接弹出。此时的stack1仍然是空的。
总结:
插入一个元素:直接将元素插入stack1即可;
删除一个元素:当stack2不为空时 ,直接弹出栈顶元素,当stack2为空时,将stack1元素逐个弹出并压入stack2,然后在弹出栈顶元素;
class Solution
{
public:
void push(int node) {
stack1.push(node);
}
int pop() {
int x;
if (stack2.empty()) //stack2为空时,将stack1元素逐个弹出并压入stack2,然后在弹出stack2的栈顶元素
{
while (!stack1.empty())
{
stack2.push(stack1.top());
stack1.pop();
}
x = stack2.top();
stack2.pop();
}
else //stack2不为空时,直接弹出stack2的栈顶元素
{
x = stack2.top();
stack2.pop();
}
return x;
}
private:
stack<int> stack1;
stack<int> stack2;
};