关闭

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

173人阅读 评论(0) 收藏 举报
分类:

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();
    }
};


0
0

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