lintcode-用栈实现队列-40

来源:互联网 发布:初级会计怎么备考 知乎 编辑:程序博客网 时间:2024/06/05 10:22

正如标题所述,你需要使用两个栈来实现队列的一些操作。

队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例

比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2

挑战

仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的

class Queue {public:    stack<int> stack1;  //进来的元素都放在里面    stack<int> stack2;  //元素都从这里出去    void push(int element) {         stack1.push(element);    }        int pop() {        if(stack2.empty()){  //如果没有元素,就把负责放入元素的栈中元素全部放进来            while(!stack1.empty()){                stack2.push(stack1.top());                stack1.pop();            }        }        int ret=stack2.top(); //有元素后就可以弹出了        stack2.pop();        return ret;    }    int top() {        if(stack2.empty()){            while(!stack1.empty()){                stack2.push(stack1.top());                stack1.pop();            }        }        return stack2.top();    }};


0 0
原创粉丝点击