牛客《剑指Offer》 用两个栈实现队列

来源:互联网 发布:unity3d 麻将 编辑:程序博客网 时间:2024/06/05 01:54

题目描述

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

思路

栈的数据是后进先出的,所以利用两个栈的配合可以实现队列先进先出的效果。
所有存入的数据,均压入栈stack1。
要取出数据时,开始stack2为空,将所有stack1中的数据取出,以此压入stack2 中,这些数据再从stack2中取出,此
class Solution{public:    void push(int node) {       stack1.push(node);    }     int pop() {        if(stack2.size()!=0){            int tmp = stack2.top();            stack2.pop();            return tmp;        }        else{            while(stack1.size()!=0){                int tmp = stack1.top();                stack1.pop();                stack2.push(tmp);            }            return pop();        }             } private:    stack<int> stack1;    stack<int> stack2;};

时便实现了先进先出。对于stack2为空时,取数据直接从stack2开始,直至stack2为空。


原创粉丝点击