【面试题】剑指Offer-7-两个栈实现一个队列

题目概述

用两个栈实现一个队列

解题思路

栈的特点是后进先出，而队列是先进先出；为了用两个栈实现一个队列，我们还是需要两个栈来实现

其中一个栈为sin，主要操纵入队；另一个栈为sout，主要用来从此出队

图解

代码实现

//两个栈实现一个队列
template<typename T>  
class QueueBTS  
{  
public:  
    void Push(const T& x)  
    {  
        sin.push(x);  
    }  
    void Pop()  
    {  
        assert(!sin.empty() || !sout.empty());  
        if (sout.empty())  
        {  
            while (!sin.empty())  
            {  
                sout.push(sin.top());  
                sin.pop();  
            }  
        }  
        sout.pop();  
    }  
  
    const T& Front()  
    {  
        assert(!sin.empty() || !sout.empty());  
        if (sout.empty())  
        {  
            while (!sin.empty())  
            {  
                sout.push(sin.top());  
                sin.pop();  
            }  
        }  
        return sout.top();  
    }  
protected:  
    stack<T> sin;//入列的栈  
    stack<T> sout;//出列的栈  
};  

题目扩展

两个队列实现一个栈

解题思路

这里我们也是，定义两个队列，第一个队列qin主要负责元素的入栈；第二个队列qout负责出栈

图解

代码实现

//两个队列实现一个栈
template<typename T>
class StackBTQ
{
public:
	void Push(const T& x)
	{
		//入栈的元素直接push到qin里面
		qin.push(x);
	}

	void Pop()
	{
		assert(!qin.empty() || !qout.empty());
		//qin队列为空
		if (qin.empty())
		{
			while (qout.size() > 1)
			{
				qin.push(qout.front());
				qout.pop();
			}
			qout.pop();
		}
		else//qin队列不为空
		{
			while (qin.size() > 1)
			{
				qout.push(qin.front());
				qin.pop();
			}
			qin.pop();
		}
	}

	T Top()
	{
		assert(!qin.empty() || !qout.empty());
		//qin队列为空
		if (qin.empty())
		{
			while (qout.size() > 1)
			{
				qin.push(qout.front());
				qout.pop();
			}
			return qout.front();
		}
		else//qin队列不为空
		{
			while (qin.size() > 1)
			{
				qout.push(qin.front());
				qin.pop();
			}
			return qin.front();
		}
	}
protected:
	queue<T> qin;
	queue<T> qout;
};