《剑指Offer》面试题-用两个栈实现队列

题目描述：

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

 

输入：

每个输入文件包含一个测试样例。
对于每个测试样例，第一行输入一个n(1<=n<=100000)，代表队列操作的个数。
接下来的n行，每行输入一个队列操作：
1. PUSH X 向队列中push一个整数x(x>=0)
2. POP 从队列中pop一个数。

 

输出：

对应每个测试案例，打印所有pop操作中从队列pop中的数字。如果执行pop操作时，队列为空，则打印-1。

 

样例输入：

3
PUSH 10
POP
POP

样例输出：

10
-1

 

题解：应该是考察STL栈应用的一道面试题，还有一点小技巧，不要用两个水杯倒来倒去（如图，明显错误做法）。用一个水杯倒就可以了。

 

代码：C的简洁 && C++的优雅。

 

C代码：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

int main(int argc, char const *argv[])
{
	int t, data, top1, top2;
	top1 = top2 = 0;
	scanf("%d", &t);
	char command[10];
	int *stack1 = (int *)malloc( (t+1) * sizeof(int));
	int *stack2 = (int *)malloc( (t+1) * sizeof(int));
	while(t--){
		scanf("%s", command);
		if(strcmp(command, "PUSH") == 0){
			scanf("%d", &data);
			stack1[top1++] = data;
		}
		if(strcmp(command, "POP") == 0){
			if(!top1 && !top2) printf("-1\n");
			else{
				if(!top2)
					while(top1)
						stack2[top2++] = stack1[--top1];
				printf("%d\n", stack2[--top2]);
			}
		}
	}
	return 0;
}

　　

 

 

C++ 代码：

 

#include <cstdio>
#include <stack>
using namespace std;
char ch[10];
class queue
{
public:
    void push(int a)
    {
        pushStack.push(a);
    }
    int pop()
    {
        int tmp = -1;
        if (popStack.empty())
        {
            if (pushStack.empty())
                return -1;
            else
            {
                while (!pushStack.empty())
                {
                    popStack.push(pushStack.top());
                    pushStack.pop();
                }
            }
        }
        tmp = popStack.top();
        popStack.pop();
        return tmp;
    }
private:
    stack<int> pushStack;
    stack<int> popStack;
};
 
int main()
{
    int n,t;
    while(scanf("%d",&n) != EOF)
    {
        queue q;
        for(int i = 0; i < n; ++i)
        {
            scanf("%s",ch);
            if(ch[1] == 'U')
            {
                scanf("%d",&t);
                q.push(t);
            }else
            {
                printf("%d\n",q.pop());
            }
        }
    }
    return 0;
}

　　

C++的 优雅写法：

#include<iostream>
#include<stdlib.h>
#include<stack>
using namespace std;

template <typename T>class CQueue
{
public:
    CQueue(void);
    ~CQueue(void);
    void appendtail(const T& node);
    T deleteHead();
private:
    stack<T> stack1;
    stack<T> stack2;

};

//构造函数
template <typename T> CQueue<T>::CQueue(void)
{
}

//析构函数
template <typename T> CQueue<T>::~CQueue(void)
{
}

//插入元素
template <typename T> void CQueue<T>::appendtail(const T& node)
{
    stack1.push(node);
}

//删除元素并返回
template <typename T> T CQueue<T>::deleteHead()
{
    if(stack2.size()<=0)
    {
        while(stack1.size()>0)
        {
            stack2.push(stack1.top());
            stack1.pop();
        }
    }
    if(stack2.size()==0)
        throw new exception("队列为空");
    T head=stack2.top();
    stack2.pop();
    return head;
}

void main()
{
    CQueue<int> queue;
    queue.appendtail(1);
    queue.appendtail(2);
    queue.appendtail(3);
    queue.appendtail(4);
    int len=4;
    while(len>0)
    {
        cout<<queue.deleteHead()<<endl;
        --len;
    }
}