suibenzhi
4/15/2018 - 6:10 AM

用两个栈实现队列.cpp

//用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

//思路:push全部推进stack1中,pop时,将stack1中的推入stack2,此时完成了倒序。pop出stack2顶部的元素即可。每次push都push入stack1,pop时,如果stack2中有数据,就依次pop出栈顶,如果没有数据,就将stack1中的数据全部pop出stack1,push入stack2,然后popstack2顶部的。
class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }
 
    int pop() {
        int top;
        if(stack2.empty())
        {
            
            while((!stack1.empty()) && (stack1.size() > 1))
            {
                stack2.push(stack1.top());
                stack1.pop();
                 
            }
            top = stack1.top();
            stack1.pop();
        }else{
            top = stack2.top();
            stack2.pop();
        }
        return top;
    }
 
private:
    stack<int> stack1;
    stack<int> stack2;
};