用栈实现队列 -LintCode

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhaohengchuan/article/details/79951456

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

样例
比如push(1), pop(), push(2), push(3), top(), pop(),你应该返回1,2和2
挑战
仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的

思路:
利用两个栈来实现队列,当push时,元素压入stack1,当进行pop或top操作时,取stack2的栈顶或出栈,若此时stack2为空,将stack1的元素出栈并压入stack2中(保证先进先出),直至stack1为空。

#ifndef C40_H
#define C40_H
#include<iostream>
#include<stack>
using namespace std;
class MyQueue {
public:
    MyQueue() {
        // do intialization if necessary
    }

    /*
    * @param element: An integer
    * @return: nothing
    */
    void push(int element) {
        // write your code here
        stack1.push(element);//向stack1中压栈
    }

    /*
    * @return: An integer
    */
    int pop() {
        // write your code here
        int val = 0;
        //如果stack2中存在元素,返回stack2的栈顶,之后出栈
        //否则,将stack1中的元素依次出栈,并压入stack2中,返回stack2的栈顶,之后出栈
        if (stack2.empty())
        {
            while (!stack1.empty())
            {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        val = stack2.top();
        stack2.pop();
        return val;
    }

    /*
    * @return: An integer
    */
    int top() {
        // write your code here
        //如果stack2中存在元素,返回stack2的栈顶
        //否则,将stack1中的元素依次出栈,并压入stack2中,之后返回stack2的栈顶
        if (stack2.empty())
        {
            while (!stack1.empty())
            {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        return stack2.top();
    }
    stack<int> stack1, stack2;
};
#endif
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页