题目1512：用两个栈实现队列

最新推荐文章于 2025-02-18 14:11:05 发布

原创最新推荐文章于 2025-02-18 14:11:05 发布 · 623 阅读

0 ·

CC 4.0 BY-SA版权

job 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法，并提供了一个C++实现示例。该方法能够在队列中进行push和pop操作，适用于需要队列功能但只有栈可用的情况。

题目描述：

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

/*
 *题目1512：用两个栈实现队列
 * */
 
#include <iostream>
#include <stack>
#include <cstdio>
#include <cstring>
 
using namespace std;
 
template <typename _type>
class MyQueue {
public :
    MyQueue ()  {}
    ~MyQueue () {
        while (!org.empty())
            org.pop();
        while (!cpy.empty())
            cpy.pop();
    }
 
    void push (const _type& val) {
        org.push (val);
    }
    const _type pop () {
        if (cpy.empty ())  {
            if (org.empty())
                return static_cast<_type>(-1);
            else {
                while (!org.empty()) {
                    cpy.push(org.top());
                    org.pop();
                }
            }
        }
        const _type res_val = cpy.top();
        cpy.pop();
        return res_val;
    }
 
private :
    stack<_type> org;
    stack<_type> cpy;
};
 
 
int main () {
    MyQueue<int> q;
    const char * PUSH = "PUSH";
    const char * POP  = "POP";
 
    char cmd[20]; 
    int n, val;
    while (scanf ("%d", &n) == 1) {
        if (n < 1 )
            continue;
        for ( int i = 0; i < n; ++i ) {
            scanf ("%s", cmd) ;
            if (strcmp(cmd, PUSH) == 0) { 
                scanf ("%d", &val) ;    
                q.push(val);
            }
            else if (strcmp(cmd, POP) == 0) {
                printf ("%d\n", q.pop());
            }
            //cout << "cmd :" << cmd << "\tval : " << val << endl;
        }
    }
 
    return 0;
}
 
/**************************************************************
    Problem: 1512
    User: 小康师傅
    Language: C++
    Result: Accepted
    Time:70 ms
    Memory:1656 kb
****************************************************************/