题目
用两个栈来实现一个队列,使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法,即保证pop操作时队列内已有元素。
数据范围: n≤1000
要求:存储n个元素的空间复杂度为O(n) ,插入与删除的时间复杂度都是O(1)
解析
栈是一种FILO的数据结构,也就是先进后出,First In Last Out,正序输入的数据通过栈弹出后会变成倒序。队列是一种FIFO的数据结构,也就是先进先出,First In First Out。而通过两个栈实现队列,就是通过两次倒序将FILO变为FIFO的过程。
例如:向栈1压入1,2;栈1弹出数据并压入栈2的顺序为2,1;栈2再弹出的顺序就是1,2,从而实现队列先进先出的效果。
本题只需要考虑清楚两个栈的多种可能存在的情形就可以做出,并不困难。
注意:原数据第1步都是压入栈1,所以不存在栈1不空,栈2也不空的初始情况。
代码
#define Max 1000
typedef struct Stack{
int data[Max+1];
int top;
}SqStack;
SqStack s1, s2; //建立顺序栈
void push(int node) {
if(s1.top == Max && s2.top != 0){ //1栈满且2栈不为空时
printf("栈已满");
}else if(s1.top == Max && s2.top == 0){ //若1栈满,2栈为空
while(s1.top != 0){ //在1栈不为空时
s2.data[++s2.top] = s1.data[s1.top--]; //将1栈的元素从栈顶压入2栈中,实现倒序
}
}
s1.data[++s1.top] = node; //若1栈没满,则先将数据全部压入1栈
}
int pop() {
if(s2.top != 0) return s2.data[s2.top--]; //当2栈不空,将2栈的数一个一个弹出
else if(s1.top == 0){ //1栈2栈都为空时,队列为空
printf("队列为空");
return 0;
}else{
while(s1.top != 0){
s2.data[++s2.top] = s1.data[s1.top--]; //1栈不为空2栈为空时,将1栈的数压入2栈
}
return s2.data[s2.top--]; //从栈顶开始返回2栈的值
}
}