题目:
232. 用栈实现队列 - 力扣(LeetCode)
这题跟我们之前写过的 用队列实现栈 很像,感兴趣的可以自行了解一下。
准备工作
这题明确说明了需要用栈来实现队列,介于C语言没有队列的库,所以在此之前我们需要用调用之前学的栈,详见栈
我们如果之前写过的话,直接复制过来就可以了,由于重点是讲用栈实现队列,所以这里就不过多了解栈了。
typedef int STDataType;
struct Stack
{
STDataType* a;
int top; // 栈顶
int capacity; // 容量,方便增容
};
//typedef struct Stack ST;
typedef struct Stack Stack;
void StackInit(Stack* pst);
void StackDestroy(Stack* pst);
// 性质就决定在栈顶出入数据
void StackPush(Stack* pst, STDataType x);
void StackPop(Stack* pst);
STDataType StackTop(Stack* pst);
bool StackEmpty(Stack* pst);
int StackSize(Stack* pst);
void StackInit(Stack* pst)
{
assert(pst);
pst->a = (STDataType*)malloc(sizeof(STDataType) * 4);
pst->top = 0;
pst->capacity = 4;
}
void StackDestroy(Stack* pst)
{
assert(pst);
free(pst->a);
pst->a = NULL;
pst->capacity = pst->top = 0;
}
// 性质就决定在栈顶出入数据
void StackPush(Stack* pst, STDataType x)
{
assert(pst);
if (pst->top == pst->capacity)
{
STDataType* tmp = (STDataType*)realloc(pst->a, sizeof(STDataType)*pst->capacity * 2);
if (tmp == NULL)
{
printf("realloc fail\n");
exit(-1); // 结束整个程序
}
pst->a = tmp;
pst->capacity *= 2;
}
pst->a[pst->top] = x;
pst->top++;
}
void StackPop(Stack* pst)
{
assert(pst);
assert(!StackEmpty(pst));
pst->top--;
}
STDataType StackTop(Stack* pst)
{
assert(pst);
assert(!StackEmpty(pst));
return pst->a[pst->top - 1];
}
bool StackEmpty(Stack* pst)
{
assert(pst);
return pst->top == 0;
}
int StackSize(Stack* pst)
{
assert(pst);
return pst->top;
}
看懂题
这样之后,我们有些小伙伴觉得自己英语不好,看不懂那些函数名,从而放弃这题,其实这里和英语好不好是没有太多关系的。注意看:
我们通过返回值大概了解了,题目中函数的含义。
创建结构体
“typedef struct”这个函数,有过对数据结构的基础认识,看到“struct”肯定是创建结构体。创建出的结构体决定了我们之下来的解题思路。所以我们先来了解一下实现这题的基本思路:
我们如此反复倒数据,每次到最后一个元素就把它pop掉,所以我们避免不了要用到两个指针,指向两个链表的指针。如此一来“typedef struct”这个函数就容易实现了。
typedef struct
{
//定义两个栈指针
Stack* pop;
Stack* push;
} MyQueue;
创建空间
“MyQueue* myQueueCreate() ”我们有了结构体,一定要初始化,不初始化会导致结构体内部的值不确定,容易出错。现在最新的“vs”如果不初始化,都编译不了了。
MyQueue* myQueueCreate()
{
MyQueue* ps = (MyQueue*)malloc(sizeof(MyQueue));
if(ps == NULL)
{
perrof("malooc:");
excit(-1);
}
//初始化
StackInit(&obj->pop);
StackInit(&obj->push);
return ps;
}
入栈
这个只要我们把push里的数据导入pop里就行了。
void myQueuePush(MyQueue* obj, int x)
{
StackPush(&obj->push, x);
}
出栈
上图分析了,这个要分有无数据两种情况讨论。
int myQueuePop(MyQueue* obj)
{
if(StackEmpty(&obj->pop))
{
//倒数据
while(!StackEmpty(&obj->push))
{
StackPush(&obj->pop, StackTop(&obj->push));
StackPop(&obj->push);
}
}
else
{
StackTop(&obj->pop);
}
}
获取元素
这里还是要分两种情况,但是我们可以取巧一点:我们假设,pop有数据的时候,我们直接取顶,或者直接调用上文写的出栈函数;pop无数据的时候,我们就倒数据,就行了。
int myQueuePeek(MyQueue* obj) {
if (StackEmpty(&obj->pop))
{
while (!StackEmpty(&obj->push))
{
StackPush(&obj->pop, StackTop(&obj->push));
StackPop(&obj->push);
}
}
return StackTop(&obj->pop);
}
这里我们再来看上文的出栈,发现这里其实可以相互调用的。
int myQueuePop(MyQueue* obj)
{
int top = myQueuePeek(obj);
StackPop(&obj->popST);
return top;
}
这样就简化了代码。
判空
两个队列同时为空
bool myQueueEmpty(MyQueue* obj)
{
return StackEmpty(&obj->push) && StackDestory(&obj->pop);
}
释放空间
释放两个链表的同时,不要忘记释放两个指针,避免内存泄漏。
void myQueueFree(MyQueue* obj)
{
StackDestory(&obj->push);
StackDestory(&obj->push);
free(ps);
ps = NULL;
}
总结
我们可以发现原理(几幅图就能搞定)都不难,主要是代码的实现,需要注意许多细节,一定要注意避免内存泄漏。
欢迎大家点赞和评论。