【面试题】用栈实现队列

题目描述

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

队列应支持push(element)，pop() 和 top()，其中pop是弹出队列中的第一个(最前面的)元素。pop和top方法都应该返回第一个元素的值。
http://www.lintcode.com/zh-cn/problem/implement-queue-by-two-stacks/

代码实现

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
        int data;

        if(stack2.empty())
        {
            while(!stack1.empty())
            {
                data = stack1.top();
                stack1.pop();
                stack2.push(data);
            }
        }

        data = stack2.top();
        stack2.pop();

        return data;

    }

    int top() {
        // write your code here
         int data;

        if(stack2.empty())
        {
            while(!stack1.empty())
            {
                data = stack1.top();
                stack1.pop();
                stack2.push(data);
            }
        }

        data = stack2.top();

        return data;
    }
};