工作生活

9.两个栈实现队列

2019-06-30  本文已影响0人  HamletSunS

思路:

class Solution
{
public:
    void push(int node) {
       stack2.push(node);
    }

    int pop() {
        if(stack1.size()!=0){
            int ret=stack1.top();
            stack1.pop();
            return ret;
        }
        else if(stack1.size()==0 && stack2.size()!=0){
            while(!stack2.empty()){
                stack1.push(stack2.top());
                stack2.pop();
            }
            int ret=stack1.top();
            stack1.pop();
            return ret;
        }
        else
            return -1;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};
上一篇 下一篇

猜你喜欢

热点阅读