用两个栈实现队列

来源:互联网 发布:手机音量减小软件 编辑:程序博客网 时间:2024/06/03 22:06

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
整体思路是元素先依次进入栈1,再从栈1依次弹出到栈2,然后弹出栈2顶部的元素,整个过程就是一个队列的先进先出 。

在交换元素的时候需要判断两个栈的元素情况: “进队列时”,队列中是否有元素,若有,说明栈2中的元素不为空,此时就先将栈2的元素倒回到栈1 中,保持在“进队列状态”。

class Solution{public:    void push(int node) {        stack1.push(node);//将数据直接压入栈1    }    int pop() {        if(stack2.empty()){//判断栈2是否为空            while(!stack1.empty()){                stack2.push(stack1.top()); //元素从栈1依次弹入,压入栈2                stack1.pop();              }        }        int result = stack2.top(); //弹出栈2栈顶元素,并返回        stack2.pop();        return result;    }private:    stack<int> stack1;    stack<int> stack2;};






原创粉丝点击