题解 | #用两个栈实现队列#

用两个栈实现队列

https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6

class Solution
{
public:
    void push(int node) {
	  //进栈入in栈
        stack1.push(node);
        
    }

    int pop() {
        int res;
        if(stack1.empty()&&stack2.empty())
            return -1;
		//如果out栈为空,将in栈内的数据全部放入out栈,此时out栈栈顶即为队首元素
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
	  //如果out栈不为空,直接out栈栈顶初战即可
        if(!stack2.empty()){
            res=stack2.top();
            stack2.pop();
            return res;
        }    
    return 0;
    }

private:
    stack<int> stack1; //in栈
    stack<int> stack2;  //out栈
};

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务