两个栈实现队列功能解析

都知道队列是先进先出,栈是先进后出。
队列打入1 2 3 4,pop出的元素应该是1才对,
反之栈打入1 2 3 4,pop出的元素是4。
所以我们可以通过循环再用一个栈去存上一个栈的栈顶,
第二个栈则为4 3 2 1,这是pop出第二个栈的栈顶就是1了。从而达到队列的先进先出功能。

源代码:

class CQueue {
public:
    CQueue() 
    {
        while(!in.empty())
        {
            in.pop();
        }
        while(!out.empty())
        {
            out.pop();
        }
    }
    void appendTail(int value) //假设这里打入1 2 3 4 
    {
        in.push(value);
    }
    
    int deleteHead() 
    {
         if(out.empty())//先判断out是否为空
        {
            if(in.empty())//如果in也为空返回-1
            {
                return -1;
            }
            else
            {
                 while(!in.empty())//循环打入out, 4 3 2 1
                {
                    out.push(in.top()); //out存入in的栈顶元素
                    in.pop();   //删除栈顶元素
                }   //将in里面的元素,从栈顶开始打入out里面
            }
        }
       
        int value = out.top();
        out.pop();//弹出栈顶1,就完成删除头部1
        return value;//返回弹出的元素
    }
private:
    stack in;
    stack out;
};

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue* obj = new CQueue();
 * obj->appendTail(value);
 * int param_2 = obj->deleteHead();
 */

作者:qi-mu-ik
链接:https://leetcode.cn/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof/solution/by-qi-mu-ik-s7rg/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(c++)