【栈和队列面试题】用栈实现队列(动图解析更清晰)

leetcode 232.用栈实现队列

前言:用两个栈实现一个队列，模拟实现队列的功能。

个人主页:​​​​​​Dream_Chaser～ 

✨✨刷题专栏:http://t.csdn.cn/UlvTc

本篇内容:力扣上栈与队列面试题目

目录

leetcode 232.用栈实现队列

结构体类型的声明(MyQueue)

1️⃣队列的初始化(myQueueCreate)

2️⃣元素入队列(myQueuePush)

3️⃣获得队首元素(myQueuePeek)

4️⃣元素出队列(myQueuePop)

5️⃣判空(myQueueEmpty)

6️⃣销毁队列(myQueueFree)

总代码:

---

来源:232. 用栈实现队列 - 力扣（LeetCode）

结构体类型的声明(MyQueue)

        使用栈来模式队列的行为，如果仅仅用一个栈，是一定不行的，所以需要两个栈一个输入栈(pushst)，一个输出栈(popst)

typedef struct {
	 ST pushst;//输入栈
 	 ST popst;//输出栈
} MyQueue;//自定义队列结构体

1️⃣队列的初始化(myQueueCreate)

图解:

代码实现: 

MyQueue* myQueueCreate() {
    //使用malloc函数为MyQueue结构体分配内存空间,定义一个MyQueue*  的指针指向这块空间
        
		MyQueue* obj =(MyQueue*)malloc(sizeof(MyQueue));
		// 若分配失败
        if(obj == NULL)
		{//则打印错误信息
			perror("malloc fail");
			return NULL;//返回NULL
		}
		STInit(&obj->pushst);//初始化输入栈
		STInit(&obj->popst);//初始化输出栈
		return obj;//返回指向这个结构体的指针
}

2️⃣元素入队列(myQueuePush)

图解：在push数据的时候，只要数据放进输入栈就好

void myQueuePush(MyQueue* obj, int x) {
		STPush(&obj->pushst , x);//往队列导入元素 
}

3️⃣获得队首元素(myQueuePeek)

        假设我们需要在正常的队列中获取头部元素，那么应该怎么在这个自定义的队列中获得呢？

        那么就应该想清楚(Output stack)输出栈的出栈顺序，由下图可以看出，输入栈(Input stack)出栈顺序是:3 2 1,那么(Output stack)输出栈的入栈顺序是: 3 2 1,自然的，(Output stack)输出栈的出栈顺序就应该为1 2 3，那么(Output stack)输出栈的这个元素1就是队列的队首元素。

        这个函数的功能，先判断输出栈为空的前提下，循环条件是输入栈不为空,STPush()的功能就是把输入栈的元素，一个一个导入到输出栈，再一个一个把输入栈的元素弹出。直到输入栈为空栈时结束循环，返回此时的输出栈栈顶元素。

int myQueuePeek(MyQueue* obj) {
		if(STEmpty(&obj->popst))//若输出栈为空
		{	
            //结束条件是把输入栈元素全部挪动到输出栈为止
			while(!STEmpty(&obj->pushst))//输入栈不为空
			{   //获得输入栈头部元素,挪动到输出栈
				STPush(&obj->popst,STTop(&obj->pushst));
				STPop(&obj->pushst);//弹出输入栈元素
			}
		}
        //返回此时的输出栈栈顶元素
		return STTop(&obj->popst);
}

4️⃣元素出队列(myQueuePop)

 错误操作:假设输入栈数据没有全部导入到输出栈里，那么最终的出栈顺序会混乱的。

正确操作:        

        所以在pop的时候，操作就复杂一些，输出栈如果为空，就把进栈数据全部导入进来（注意是全部导入），再从出栈弹出数据，如果输出栈不为空，则直接从出栈弹出数据就可以了。

代码实现的部分需要注意的： 

        可以看出myQueuePop的实现，直接复用了myQueuePeek() ,要不然，对stEmpty判空的逻辑又要重写一遍。

代码实现: 

int myQueuePop(MyQueue* obj) {
		int front = myQueuePeek(obj);//代码复用,获得队列头部元素
		STPop(&obj->popst);//弹出元素
		return front;//获得队列的头部元素
}

5️⃣判空(myQueueEmpty)

如果进栈和出栈都为空的话，说明模拟的队列为空了。

bool myQueueEmpty(MyQueue* obj) {
		return STEmpty(&obj->pushst) 
		&&	STEmpty(&obj->popst);
}//判断队列是否为空，需要输入栈和输出栈两个同时为空，才会返回true

6️⃣销毁队列(myQueueFree)

和队列实现栈同样道理，需要先释放两个栈的空间，再释放结构体的空间，这样才确保不会内存泄露

//队列空间的释放
void myQueueFree(MyQueue* obj) {
		STDestroy(&obj->pushst);//释放输入栈
		STDestroy(&obj->popst);//释放输出栈
		free(obj);//释放自定义的队列
}

总代码:

#include
#include
#include
#include
typedef int STDataType;
typedef struct Stack
{
	STDataType* a;
	int top;//栈顶的位置
	int capacity;//栈的容量
}ST;
 
void STInit(ST* pst);
void STDestroy(ST* pst);
void STPush(ST* pst,STDataType x);
void STPop(ST* pst);
STDataType STTop(ST* pst);
bool  STEmpty(ST* pst);
int STSize(ST*pst);


void STInit(ST* pst)
{
	assert(pst);
	pst->a = NULL;//栈底
	
	//top不是下标
    //pst->top=-1;//指向栈顶元素
	pst->top = 0;//指向栈顶元素的下一个位置
	pst->capacity = 0;
}
 
void STDestroy(ST* pst)
{
	assert(pst);
	free(pst->a);
	pst->a = NULL;
}
 
 
void STPush(ST* pst,STDataType x)
{
	if (pst->top == pst->capacity)
	{
		int newCapacity = pst->capacity == 0 ? 4 : pst->capacity * 2;//true,4.false,括2倍
 
		STDataType* tmp = (STDataType*)realloc(pst->a, newCapacity * sizeof(STDataType));//返回值地址相等就是原地扩容，不同就是异地扩
		if (tmp == NULL)
		{
			perror("realloc fail");
			return;
		}
		pst->a = tmp;//返回的是realloc出来的内存块的地址
		pst->capacity = newCapacity;//把扩容后的空间大小赋值给栈容量
	}
	pst->a[pst->top] = x;//先放值
	pst->top++;//再++
}
 
 
void STPop(ST* pst)
{
	assert(pst);
	assert(!STEmpty(pst));
	pst->top--;
}
 
 
STDataType STTop(ST* pst)
{
	assert(pst);
	assert(!STEmpty(pst));
 
	return pst->a[pst->top - 1];
}
 
bool STEmpty(ST* pst)//栈为空返回true,不为空返回false
{
	//assert(pst);
	//if (pst->top == 0)
	//{
	//	return true;
	//}
	//else
	//{
	//	return false;
	//}
	return pst->top == 0;
}

int STSize(ST* pst)
{
	assert(pst);
	return pst->top;
}
 


typedef struct {
	 ST pushst;//输入栈
 	 ST popst;//输出栈
} MyQueue;//自定义队列

MyQueue* myQueueCreate() {
    //使用malloc函数为MyQueue结构体分配内存空间,定义一个MyQueue*  的指针指向这块空间
        
		MyQueue* obj =(MyQueue*)malloc(sizeof(MyQueue));
		// 若分配失败
        if(obj == NULL)
		{//则打印错误信息
			perror("malloc fail");
			return NULL;//返回NULL
		}
		STInit(&obj->pushst);//初始化输入栈
		STInit(&obj->popst);//初始化输出栈
		return obj;//返回指向这个结构体的指针
}

void myQueuePush(MyQueue* obj, int x) {
		STPush(&obj->pushst , x);//往队列导入元素 
}

int myQueuePop(MyQueue* obj) {
		int front = myQueuePeek(obj);//代码复用,获得队列头部元素
		STPop(&obj->popst);//弹出元素
		return front;//获得队列的头部元素
}


int myQueuePop(MyQueue* obj) {
	if(STEmpty(&obj->popst))//若输出栈为空
		{	
            //结束条件是把输入栈元素全部挪动到输出栈为止
			while(!STEmpty(&obj->pushst))//输入栈不为空
			{   //获得输入栈头部元素,挪动到输出栈
				STPush(&obj->popst,STTop(&obj->pushst));
				STPop(&obj->pushst);//弹出输入栈元素
			}
		}
        //返回此时的输出栈头部元素
		int front= STTop(&obj->popst);
		STPop(&obj->popst);//弹出元素
		return front;//获得队列的头部元素
}

int myQueuePeek(MyQueue* obj) {
		if(STEmpty(&obj->popst))//若输出栈为空
		{	
            //结束条件是把输入栈元素全部挪动到输出栈为止
			while(!STEmpty(&obj->pushst))//输入栈不为空
			{   //获得输入栈头部元素,挪动到输出栈
				STPush(&obj->popst,STTop(&obj->pushst));
				STPop(&obj->pushst);//弹出输入栈元素
			}
		}
        //返回此时的输出栈头部元素
		return STTop(&obj->popst);
}

bool myQueueEmpty(MyQueue* obj) {
		return STEmpty(&obj->pushst) 
		&&	STEmpty(&obj->popst);
}//判断队列是否为空，需要输入栈和输出栈两个同时为空，才会返回true

//队列空间的释放
void myQueueFree(MyQueue* obj) {
    //和队列实现栈同样道理，需要先释放两个栈的空间，再释放结构体的空间，这样才确保不会内存泄露
		STDestroy(&obj->pushst);//释放输入栈
		STDestroy(&obj->popst);//释放输出栈
		free(obj);//释放自定义的队列
}

代码执行: 

        本文结束，如有错误，欢迎指正，感谢你的来访！