【剑指Offer面试编程题】题目1512：用两个栈实现队列--九度OJ

题目描述：

用两个栈来实现一个队列，完成队列的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

【解题思路】新建栈st1模拟队列的输入，只要遇到push操作，都将元素压入st1中。另外，新建栈st2模拟队列的输出，如果pop操作时，先判断st1是否为空，当st2为空，则将st1的中的所有元素挨个弹出并压入st2中以备输出，这时的st2中的元素的排列正好是当前所有元素的输入的逆序，符合队列的输出顺序；-->>当出现pop操作时，如果st2不为空，这个时候的栈顶元素真好是当前状态的队列的头元素，则直接弹出栈顶元素作为队列的输出。

AC code：

#include <cstdio>
#include <stack>
using namespace std;
 
int main()
{
  int n;
  scanf("%d",&n);
  stack<int> st1,st2;
  for(int i=0;i<n;++i)
  {
    char cc[5];
    int tt;
    scanf("%s",cc);
    if(cc[1]=='U')
    {
      scanf("%d",&tt);
      st1.push(tt);
    }else
    {
      if(st2.empty())
      {
        while(!st1.empty())
        {
          int te=st1.top();
          st2.push(te);
          st1.pop();
        }
      }
      if(st2.empty())
      {
        printf("-1
");
        continue;
      }
      int te=st2.top();
      printf("%d
",te);
      st2.pop();
    }
  }
  return 0;
}
/**************************************************************
    Problem: 1512
    User: huo_yao
    Language: C++
    Result: Accepted
    Time:60 ms
    Memory:1656 kb
****************************************************************/

题目链接：http://ac.jobdu.com/problem.php?pid=1512

【剑指Offer面试编程题】题目1512：用两个栈实现队列--九度OJ

相关推荐