关闭

leetcode 232 Implement Queue using Stacks C++

标签: LeetCode
384人阅读 评论(0) 收藏 举报
分类:

用两个栈实现一个队列。

class Queue {
public:
    stack<int> s1;
    stack<int> s2;
    // 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.empty()) s2.pop();
        else{
            if(!s1.empty()){
                while(s1.size()){
                    s2.push(s1.top());
                    s1.pop();
                }
                s2.pop();
            }
        }
    }

    // Get the front element.
    int peek(void) {
        if(!s2.empty()) return s2.top();
        else{
            if(!s1.empty()){
                while(s1.size()){
                    s2.push(s1.top());
                    s1.pop();
                }
                return s2.top();
            }else{
                return NULL;
            }
        }
    }

    // Return whether the queue is empty.
    bool empty(void) {
        
        return s1.empty() && s2.empty();
    }
};


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:68887次
    • 积分:2459
    • 等级:
    • 排名:第15615名
    • 原创:178篇
    • 转载:42篇
    • 译文:0篇
    • 评论:13条
    文章分类
    最新评论