用两个栈实现队列

用两个栈实现队列

http://www.nowcoder.com/questionTerminal/54275ddae22f475981afa2244dd448c6

首先是一些函数
1.判断堆栈为空:bool empty();
2.移除堆栈最顶层元素:void pop();
3.将val 值压栈:void push(const TYPE &val);
4.当前栈中元素数目 size_type size();
5.返回对栈顶元素的引用 TYPE &top();
一、两个栈为空时
向stack1弹入数字,先进后出,依次弹出后存入第二个栈,从第二个栈弹出时即为模拟出来的先进先出。
eg.向stack1中依次存入1.2.3,4有stack1={1,2,3,4}弹出顺序为4,3,2,1再依次存入stack2={4,3,2,1}
可首先弹出1 stack2.pop() 之后为stack2={4,3,2}
二、当stack1为空,stack2不为空时插入数据5
eg.stack1={},stack2={4,3,2}
插入数据5时,应该弹出值为2即stack2栈顶元素。
可知此时的插入处理应为stack1.push(node),然后弹出stack2.pop()
三、当stack2为空时,插入元素6
eg.stack1={5},stack2={}
插入stack1有stack1={5,6}仅当stack2为空时,将stack1的元素依次弹出,存入stack2.

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

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

private:
stack<int> stack1;
stack<int> stack2;
};</int></int>

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务