码迷,mamicode.com
首页 > 其他好文 > 详细

两个栈实现一个队列

时间:2015-07-20 21:37:21      阅读:101      评论:0      收藏:0      [点我收藏+]

标签:

用两个栈实现一个队列的功能。

解题思路

假设两个栈A和B,且都为空。
栈A提供push()功能,栈B提供pop()功能。

  • 入队列:入栈A。
  • 出队列:
    • 如果栈B不为空,直接弹出B的元素。
    • 如果栈B为空,则依次弹出栈A的元素并压入栈B中,再弹出B中的元素。

实现代码

#include <iostream>
#include <stack>
using namespace std;

template<class T>
class MyQueue
{
public:
    void push(const T& t)
    {
        s1.push(t);
    }

    void pop()
    {
        if (s2.empty())
        {
            while (!s1.empty())
            {
                s2.push(s1.top());
                s1.pop();
            }
        }
        s2.pop();
    }

    T& top()
    {
        if (s2.empty())
        {
            while (!s1.empty())
            {
                s2.push(s1.top());
                s1.pop();
            }
        }

        return s2.top();
    }

    bool empty() const
    {
        return s1.empty() && s2.empty();
    }

private:
    stack<T> s1;
    stack<T> s2;
};


int main()
{
    MyQueue<int> myque;
    for (int i = 0; i < 10; i++)
    {
        myque.push(i);
    }

    while (!myque.empty())
    {
        cout<<myque.top()<<" ";
        myque.pop();
    }
}

注意:pop()top()中没有再加上if (!s2.empty())的判断,我认为此处应该由使用者来判断MyQueue对象是否为空,减少一些不必要的判断可以提高程序效率。

版权声明:本文为博主原创文章,未经博主允许不得转载。

两个栈实现一个队列

标签:

原文地址:http://blog.csdn.net/foreverling/article/details/46973171

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!