《剑指Offer》读书笔记05:用两个栈实现队列

来源:互联网 发布:测评软件 编辑:程序博客网 时间:2024/06/06 04:30

题目描述

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


解题思路

  1. 队列的特点先进先出,栈的特点后进先出。
  2. 假如序列a,b,c压入栈stack1,在stack1中出栈,以顺序c,b,a压入stack2,那么stack2出栈时,顺序变为a,b,c,基本满足队列要求。
  3. 在步骤2中,假如stack2中的a,b出栈,留下c,此时在stack1中压入d,e。此时能不能把d,e从stack1中弹出,压入stack2呢?从整个入栈顺序来看,下一个出栈的应该是c,若e,d压入stack2,下一个出栈的是d,不符合队列弹出顺序。
  4. 从步骤3可以得出,stack2中有元素时,先弹出。当stack2为空时,才能从stack1弹出元素压入stack2.

参考代码

class Solution{public:    void push(int node) {       stack1.push(node);     }    int pop() {        if ( stack2.empty() ) {//stack2为空时,才从stack1压入元素            while ( !stack1.empty() ) {                int tmp = stack1.top();//top()为取栈顶元素值                stack2.push(tmp);                stack1.pop();//pop()为弹出(删除)栈顶元素。                //top、pop容易傻傻分不清楚            }        }        if ( stack2.empty() ) {            //throw new exception("Queue is empty!");        }        int theResult = stack2.top();        stack2.pop();        return theResult;    }private:    stack<int> stack1;    stack<int> stack2;};
0 0
原创粉丝点击