数据结构和算法分析程序员

lintcode 用栈实现队列

2016-12-26  本文已影响174人  yzawyx0220

正如标题所述,你需要使用两个栈来实现队列的一些操作。
队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。
pop和top方法都应该返回第一个元素的值。
样例
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2

设置两个栈,stack1和stack2,push操作不用说直接push就可以。对于pop操作,因为栈是先进后处,队列是先进先出,因此要把栈里的数反转一下,这样就只需要从stack1的栈顶依次压入stack2,然后返回stack.top()即可。top操作和pop是一样的。

class Queue {
public:
    stack<int> stack1;
    stack<int> stack2;

    Queue() {
        // do intialization if necessary
    }

    void push(int element) {
        // write your code here
        stack1.push(element);
    }
    
    int pop() {
        // write your code here
        if (stack2.empty()){
            while (!stack1.empty()) {
                int temp = stack1.top();
                stack2.push(temp);
                stack1.pop();
            }
        }
        int res = stack2.top();
        stack2.pop();
        return res;
    }

    int top() {
        // write your code here
        if (stack2.empty()) {
            while (!stack1.empty()) {
                int temp = stack1.top();
                stack2.push(temp);
                stack1.pop();
            }
        }
        return stack2.top(); 
    }
};

上一篇下一篇

猜你喜欢

热点阅读