面试题六 用两个栈实现队列

题目

　　用两个栈实现队列的进队和出队功能

分析

　　1. 栈直接用C++的容器适配器 stack 实现

　　2. 当进队时，直接往栈 1 存放数据；当出队时，如果栈 2 非空则直接从栈 2 取数，否则将栈 1 的数据全部出栈并转移至栈 2，然后再从栈 2 取数。

　　3. 考虑到这里采用的两个栈应当要封装起来，因此，采用类来实现这个队列。

代码实现

#include <iostream>
#include <stack>

using namespace std;

// 队列定义
class SQueue {
public:
    SQueue (void) {
    }
    ~SQueue (void) {
    }
    // 入队函数
    void appendTail (const int n);
    // 出队函数
    int deleteHead();
    // 队判空函数
    bool empty();
private:
    stack <int> s1; // 使用容器适配器实现栈
    stack <int> s2;
};

void SQueue :: appendTail (const int n) {
    s1.push (n);
}

int SQueue :: deleteHead() {
    int tem;
    // 首先将第二个栈的元素出栈，如果第二个栈中没有元素可出了，那么就将第一个栈中的元素转移到第二个栈中来。
    if (s2.size() == 0) {
        while (s1.size() != 0) {
            tem = s1.top();
            s1.pop();
            s2.push(tem);
        }
    }

    tem = s2.top();
    s2.pop();

    return tem;
}

bool SQueue :: empty() {
    // 两个栈都为空则说明队列为空
    if (s1.size() == 0 && s2.size() ==0) 
        return 1;
    else return 0;
}

int main (void) {
    SQueue q;

    // 将 9 5 4 7 这四个整数进队
    q.appendTail(9);
    q.appendTail(5);
    q.appendTail(4);
    q.appendTail(7);

    // 让之前进队的四个数出队并打印
    int num;
    while (!q.empty()) {
        num = q.deleteHead();
        cout << num << " ";
    }
    cout << endl;

    return 0;
}

小结

　　1. 本题只要求实现入队和出队函数，其实可以继续添加机制完善这个队列：如复制控制，空间限制等。