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

来源:互联网 发布:mac下的思维导图 编辑:程序博客网 时间:2024/05/17 01:54
题目描述:

用两个栈来实现一个队列,完成队列的Push和Pop操作。
队列中的元素为int类型。

输入:

每个输入文件包含一个测试样例。
对于每个测试样例,第一行输入一个n(1<=n<=100000),代表队列操作的个数。
接下来的n行,每行输入一个队列操作:
1. PUSH X 向队列中push一个整数x(x>=0)
2. POP 从队列中pop一个数。

输出:

对应每个测试案例,打印所有pop操作中从队列pop中的数字。如果执行pop操作时,队列为空,则打印-1。

样例输入:

3PUSH 10POPPOP
样例输出:

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

0 0
原创粉丝点击