题目描述

• 用两个栈实现一个队列，完成队列的 push 和 pop 操作
• 队列的元素为 int 类型

解题思路

• 剑指Offer 69

代码

• 第二轮代码

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

    int pop() {
        int tt;
        if(!stack2.empty()){// 如果stack2不为空，直接返回stack2栈顶元素
            tt = stack2.top();
            stack2.pop();
        }
        else{ // stack2 为空,先把stack1放入到stack2再返回
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
            tt = stack2.top();
            stack2.pop();
        }
        return tt;
    }

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

我是分割线

• 第一轮代码

#include<iostream>
#include<stack>
using namespace std;

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

    int pop() {
        int temp;
        if(!stack2.empty()){
            temp = stack2.top();
            stack2.pop();
        }
        else{
            int tt = stack1.size()-1;
            for(int i=0; i<tt; i++){
                temp = stack1.top();
                stack2.push(temp);
                stack1.pop();
            }
            temp = stack1.top();
            stack1.pop();
        }
        return temp;
    }

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

main(){
    Solution ss;

    ss.push(1);
    ss.push(2);
    ss.push(3);
    cout<<ss.pop();
    cout<<ss.pop();
    ss.push(4);
    cout<<ss.pop();
    ss.push(5);
    cout<<ss.pop();
    cout<<ss.pop();
}
//  输出为: 12345

总结展望

• 题目相当不错，引人深思!!!

参考

• C++ 栈和队列的介绍与使用