LeetCode || Implement Queue using Stacks

来源:互联网 发布:类似于完美世界 知乎 编辑:程序博客网 时间:2024/05/01 20:02
class Queue {public:    // Push element x to the back of queue.    void push(int x) {        s1.push(x);    }    // Removes the element from in front of queue.    void pop(void) {        if(s2.size() == 0)        {            while(s1.size()!=0)            {                s2.push(s1.top());                s1.pop();            }        }        s2.pop();    }    // Get the front element.    int peek(void) {        if(s2.size() == 0)        {            while(s1.size()!=0)            {                s2.push(s1.top());                s1.pop();            }        }        return s2.top();    }    // Return whether the queue is empty.    bool empty(void) {        if(s1.size() == 0 && s2.size() == 0)            return true;        return false;    }        stack<int> s1;    stack<int> s2;};

0 0
原创粉丝点击