题目描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
解题思路
我们定义两个栈stack1和stack2,stack1用来插入元素,当要删除元素时,首先将stack1中的所有元素全部push到stack2中,这样stack1栈顶的元素就到了stack2栈底,stack1栈底的元素就到了stack2栈顶,再将其出栈。只有当stack2为空时再将stack1中的元素push到stack2中,这样可以防止数据乱序。
代码实现
class myQueue
{
public:
void appendTail(int x);
int deleteHead();
bool empty();
private:
stack<int> stack1;
stack<int> stack2;
};
void myQueue::appendTail(int x)
{
stack1.push(x);
}
int myQueue::deleteHead()
{
if (stack2.empty())
{
while (!stack1.empty())
{
stack2.push(stack1.top());
stack1.pop();
}
}
int ret = stack2.top();
stack2.pop();
return ret;
}
bool myQueue::empty()
{
if (stack1.empty() && stack2.empty())
return false;
return true;
}