【剑指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\n");

        continue;

      }

      int te=st2.top();

      printf("%d\n",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

最新文章

热门文章