利用两个栈实现一个队列(C++版)

题目：

利用两个栈实现一个队列的功能，完成pop() push() peek() empty()等功能。

push(x) -- Push element x to the back of queue.

pop() -- Removes the element from in front of queue.

peek() -- Get the front element.

empty() -- Return whether the queue is empty.

主要思想：一个栈作为压入栈(只执行push操作)，另一个栈作为弹出栈(只执行pop操作)。

push —— 执行压入操作时，将数据压入 压入栈。

pop —— 执行弹出操作时，必须将压入栈所有的数据压入弹出栈，然后取出弹出栈顶元素。

peek —— 只获取弹出栈栈顶元素 不弹出。

empty —— 只有两个栈都为空才返回true

实现函数代码如下：

class TwoStackQueue {
public:
// Push element x to the back of queue.
void push(int x) {
stackPush.push(x);
}

// Removes the element from in front of queue.
void pop(void) {
if(stackPop.empty() && !stackPush.empty())
{
while(!stackPush.empty())
{
stackPop.push(stackPush.top());
stackPush.pop();
}
}
stackPop.pop();
}

// Get the front element.
int peek(void) {
if(stackPop.empty() && !stackPush.empty())
{
while(!stackPush.empty())
{
stackPop.push(stackPush.top());
stackPush.pop();
}
}

return stackPop.top();
}

// Return whether the queue is empty.
bool empty(void) {
if(stackPush.empty() && stackPop.empty() )
{
return true;
}

return false;
}
public:
stack<int> stackPush;
stack<int> stackPop;
};