LintCode 用栈实现队列

正如标题所述,你需要使用两个栈来实现队列的一些操作。

队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例

比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2

分析:就是用两个栈实现先进先出

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 data=stack1.top();
                stack2.push(data);
                stack1.pop();
            }
            
        }
        int res=stack2.top();
        stack2.pop();
        return res;
    }

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

  

posted @ 2016-12-05 10:27  lelelelele  阅读(188)  评论(0编辑  收藏  举报