用两个栈实现队列

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

首先定义两个栈来模拟队列操作,stack1用来实现队列的入队操作,stack2来实现队列的出队操作。

首先stack2是空。

入队:直接把元素push stack1即可。出队:由于队列是先进先出,因此要把stack1栈底元素提取出来,直接一次pop stack2元素push进stack2,直到stack1为空,stack1的栈底元素就为stack2的栈顶元素,出队元素为stack2.top()。

stack2非空,出队元素为stack2.top()。

首先入队进入stack1。step1:执行出队操作,由于stack2空,依次把stack1栈顶元素压入stack2,最后stack2栈顶即为出队元素


step2:之后再进行出队操作的话,依次pop stack2栈顶元素即可。直至stack2为空,然后出队操作采用step1.

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

    int pop() {
        int a;
        if(stack2.empty())
        {
            while(!stack1.empty())
            {
                a=stack1.top();
                stack2.push(a);
                stack1.pop();
            }
        }
        a=stack2.top();
        stack2.pop();
        return a;
    }

private:
    stack<int> stack1;//入队
    stack<int> stack2;//出队
};

阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭