[九度OnlineJudge][剣指Offer]タイトル1512:2つのスタックでキューを実現

1829 ワード

タイトルの説明:
2つのスタックで1つのキューを実現し、キューのPushとPop操作を完了します.キュー内の要素はintタイプです.
入力:
各入力ファイルには、テストサンプルが含まれています.各テストサンプルについて、最初の行には、キュー操作の個数を表すn(1<=n<=10000000)が入力される.次の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 <stack>
#include <string>
#include <cstdio>
using namespace std;
 
int main()
{
    //freopen("input.txt","r",stdin);
    int n;
    cin>>n;
    string operation;
    int data;
    stack<int> stack1;
    stack<int> stack2;
    for(int i=0;i<n;i++)
    {
        cin>>operation;
        if(operation=="PUSH")
        {
            cin>>data;
            stack1.push(data);
        }
        else
        {
            if(stack2.empty())
            {
                while(!stack1.empty())
                {
                    stack2.push(stack1.top());
                    stack1.pop();
                }
                if(stack2.empty())
                    cout<<-1<<endl;
                else
                {
                    cout<<stack2.top()<<endl;
                    stack2.pop();
                }
            }
            else
            {
                cout<<stack2.top()<<endl;
                stack2.pop();
            }
        }
    }
    return 0;
}
 
 
/**************************************************************
    Problem: 1512
    User: fuestck
    Language: C++
    Result: Accepted
    Time:580 ms
    Memory:1656 kb
****************************************************************/