剑指offer中题目:http://ac.jobdu.com/problem.php?pid=1512
-
题目描述:
-
用两个栈来实现一个队列,完成队列的Push和Pop操作。
队列中的元素为int类型。
-
输入:
-
每个输入文件包含一个测试样例。
对于每个测试样例,第一行输入一个n(1<=n<=100000),代表队列操作的个数。
接下来的n行,每行输入一个队列操作:
1. PUSH X 向队列中push一个整数x(x>=0)
2. POP 从队列中pop一个数。
-
输出:
-
对应每个测试案例,打印所有pop操作中从队列pop中的数字。如果执行pop操作时,队列为空,则打印-1。
-
样例输入:
-
3 PUSH 10 POP POP
-
样例输出:
-
10 -1
代码:
#include <iostream>
#include <string>
#include <stack>
class CQueue
{
public:
CQueue() {}
~CQueue() {}
void push(int vData)
{
m_InQue.push(vData);
}
int pop()
{
if (m_OutQue.empty())
{
if (m_InQue.empty()) return -1;
__move();
}
int Data = m_OutQue.top();
m_OutQue.pop();
return Data;
}
private:
std::stack<int> m_InQue;
std::stack<int> m_OutQue;
void __move()
{
while (!m_InQue.empty())
{
m_OutQue.push(m_InQue.top());
m_InQue.pop();
}
}
};
int main()
{
int N;
std::string OperSrt;
CQueue Que;
std::cin >> N;
while (N--)
{
std::cin >> OperSrt;
if (OperSrt == "PUSH")
{
int Data;
std::cin >> Data;
Que.push(Data);
}
else
{
std::cout << Que.pop() << std::endl;
}
}
return 0;
}
/**************************************************************
Problem: 1512
User:
Language: C++
Result: Accepted
Time:510 ms
Memory:1652 kb
****************************************************************/