用两个栈实现队列功能【剑指offer】

题目描述：

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

程序代码：

【方法一】

class Solution
{
public:

    void push(int node) {
        stack1.push(node);


    }

    int pop() {
        int a;
        if (stack2.size() <= 0){
            while (stack1.size()>0){
                a = stack1.top();
                stack2.push(a);
                stack1.pop();
            }
        }
        a = stack2.top();
        stack2.pop();

        return a;


    }

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

【方法二】

class Solution
{
public:
    int cou = 0;
    void push(int node) {
        stack1.push_back(node);
        stack2.push_back(cou++);
    }

    int pop() {
        int i = 0;
        while (stack2[i] == -1)
        {
            i++;
        }
        stack2[i] = -1;
        return stack1[i];
    }

private:
    vector <int> stack1;//存数
    vector <int> stack2;//地址
};

测试用例：

int main()
{
    Solution fun;
    int a[10] = {1,2,3,0,0,4,0,5,0,0};
    int i;
    for ( i = 0; i < 10; i++)
    {
        fun.push(a[i]);
    }
    for ( i = 0; i < 10; i++)
    {
        cout << " " << fun.pop() /*<< endl*/;
    }
    getchar();
    return 0;
}