leetcode 日经贴,Cpp code -Implement Queue using Stacks

Implement Queue using Stacks

class Queue {
    stack<int> st[2];
    int inidx, outidx;
    void move(int from, int to) {
        while (!st[from].empty()) {
            st[to].push(st[from].top());
            st[from].pop();
        }
    }
public:
    Queue() {
        inidx = 0;
        outidx = 1;
    }
    // Push element x to the back of queue.
    void push(int x) {
        st[inidx].push(x);
    }

    // Removes the element from in front of queue.
    void pop(void) {
        if (st[outidx].empty() && !st[inidx].empty()) {
            move(inidx, outidx);
        }
        if (!st[outidx].empty()) {
            st[outidx].pop();
        }
    }

    // Get the front element.
    int peek(void) {
        if (st[outidx].empty() && !st[inidx].empty()) {
            move(inidx, outidx);
        }
        if (!st[outidx].empty()) {
            return st[outidx].top();
        } else {
            return -1;
        }
    }

    // Return whether the queue is empty.
    bool empty(void) {
        return st[inidx].empty() && st[outidx].empty();
    }
};


阅读更多
个人分类: 算法 leetcode Cpp
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

leetcode 日经贴,Cpp code -Implement Queue using Stacks

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭