[算法] 使用两个栈实现队列

发布时间:2019年09月07日 阅读:246 次

http://www.sohu.com/a/270671146_478315


使用STL中的stack创建两个栈,s1和s2,s1负责入栈,s2负责出栈;


队列入队:使用STL中的stack会动态分配内存,因此不必担心栈满的情况,所以入队时,直接将元素入s1栈即可。


队列出队:出队时需要s2出栈,因此要判定s2是否为空;若s2为空,则将s1中元素一一出栈进入s2栈中,此时s2已不为空;那么队列出队只需将s2出栈即可。当然,如果s1和s2均为空,则出队失败。


#include<stack>
using namespace std;

class MyQueue
{
public:
    stack<int> s1;
    stack<int> s2;

    //入队:直接s1进栈即可
    bool EnQueue(int elem)
    {
        s1.push(elem);
        return true;
    }

    //出队:若s2为空,需将s1中的元素一一出栈进入s2中,再将s2出栈
    //      若s2本身不为空,直接将s2出栈
    bool DeQueue()
    {
        if(s1.empty() && s2.empty())
        {
            return false;
        }
        if(s2.empty())
        {
            while(!s1.empty())
            {
                s2.push(s1.top());
                s1.pop();
            }
        }
        if(!s2.empty())
        {
            s2.pop();
            return true;
        }
    }
};
————————————————
版权声明:本文为CSDN博主「ColiYin」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/sinat_20177327/article/details/77712847


Tag:
相关文章

发表评论: