欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

lintcode40. 用栈实现队列

程序员文章站 2022-07-15 12:13:09
...

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

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

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

样例
例1:

输入:
    push(1)
    pop()    
    push(2)
    push(3)
    top()    
    pop()     
输出:
    1
    2
    22:

输入:
    push(1)
    push(2)
    push(2)
    push(3)
    push(4)
    push(5)
    push(6)
    push(7)
    push(1)
输出:
[]
挑战
仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的

注意事项
假设调用pop()函数的时候,队列非空
class MyQueue {
public:
    MyQueue() {
        // do intialization if necessary
    }

    /*
     * @param element: An integer
     * @return: nothing
     */
    void push(int element) {
        // write your code here
        while(!a.empty())
        {
            b.push(a.top());
            a.pop();
        }
        a.push(element);
        while(!b.empty())
        {
            a.push(b.top());
            b.pop();
        }
    }

    /*
     * @return: An integer
     */
    int pop() {
        // write your code here
        int result=a.top();
        a.pop();
        return result;
    }

    /*
     * @return: An integer
     */
    int top() {
        // write your code here
        return a.top();
        
    }
    stack<int> a;
    stack<int> b;
};