我们已经实现了顺序队列和链式队列,并且也做了优化,
那这篇文章到底是干嘛呢!!简单四个字,代码复用!!!
注:这里我们要包含链式链表和链式栈的头文件和实现文件,具体代码前几篇文章里面找。
第十二个例子,代码复用队列的实现:
头文件
#ifndef _SQUEUE_H_
#define _SQUEUE_H_
typedef void SQueue;
SQueue* SQueue_Create();
void SQueue_Destroy(SQueue* queue);
void SQueue_Clear(SQueue* queue);
int SQueue_Append(SQueue* queue, void* item);
int SQueue_Length(SQueue* queue);
void* SQueue_Retrieve(SQueue* queue);
void* SQueue_Header(SQueue* queue);
#endif
我个人有点小小的强迫症,代码尽量要求简洁,所以没有注释,有什么不明白的可以留言。
实现文件
#include "SQueue.h"
#include "LinkStack.h"
#include <malloc.h>
typedef struct tag_SQueue
{
LinkStack* inStack;
LinkStack* outStack;
}TSQueue;
SQueue* SQueue_Create()
{
TSQueue* ret = (TSQueue*)malloc(sizeof(TSQueue));
if (ret)
{
ret->inStack = LinkStack_Create();
ret->outStack = LinkStack_Create();
if ((ret->inStack == NULL) || (ret->outStack == NULL))
{
if (ret->inStack != NULL)
{
LinkStack_Destroy(ret->inStack);
}
if (ret->outStack != NULL)
{
LinkStack_Destroy(ret->outStack);
}
free(ret);
ret = NULL;
}
}
return ret;
}
void SQueue_Destroy(SQueue* queue)
{
SQueue_Clear(queue);
free(queue);
}
void SQueue_Clear(SQueue* queue)
{
TSQueue* sQueue = (TSQueue*)queue;
if (sQueue != NULL)
{
LinkStack_Clear(sQueue->inStack);
LinkStack_Clear(sQueue->outStack);
}
}
int SQueue_Append(SQueue* queue, void* item)
{
TSQueue* sQueue = (TSQueue*)queue;
if (sQueue != NULL)
{
LinkStack_Push(sQueue->inStack, item);
}
}
int SQueue_Length(SQueue* queue)
{
TSQueue* sQueue = (TSQueue*)queue;
int ret = -1;
if (sQueue != NULL)
{
ret = LinkStack_Length(sQueue->inStack) + LinkStack_Length(sQueue->outStack);
}
return ret;
}
void* SQueue_Retrieve(SQueue* queue)
{
TSQueue* sQueue = (TSQueue*)queue;
void* ret = NULL;
if (sQueue != NULL)
{
if (LinkStack_Length(sQueue->outStack) == 0)
{
while (LinkStack_Length(sQueue->inStack) > 0)
{
LinkStack_Push(sQueue->outStack, LinkStack_Pop(sQueue->inStack));
}
}
ret = LinkStack_Pop(sQueue->outStack);
}
return ret;
}
void* SQueue_Header(SQueue* queue)
{
TSQueue* sQueue = (TSQueue*)queue;
void* ret = NULL;
if (sQueue != NULL)
{
if (LinkStack_Length(sQueue->outStack) == 0)
{
while (LinkStack_Length(sQueue->inStack) > 0)
{
LinkStack_Push(sQueue->outStack, LinkStack_Pop(sQueue->inStack));
}
}
ret = LinkStack_Top(sQueue->outStack);
}
return ret;
}
这里我们全部是复用前面写好的代码,核心思想是,用2个栈,一个栈只负责入
栈,另一个只负责出栈,把两个栈看成一个整体,就跟队列效果一样,先进先出。
测试文件
#include <stdio.h>
#include <stdlib.h>
#include "SQueue.h"
int main(int argc, char *argv[])
{
SQueue* queue = SQueue_Create();
int a[10] = {0};
int i = 0;
for (i = 0; i < 10; i++)
{
a[i] = i + 1;
SQueue_Append(queue, a + i);
}
printf("length:%d\n", SQueue_Length(queue));
printf("header:%d\n", *((int*)SQueue_Header(queue)));
while (SQueue_Length(queue) > 0)
{
printf("retrieve:%d\n", *((int*)SQueue_Retrieve(queue)));
}
SQueue_Clear(queue);
SQueue_Destroy(queue);
system("PAUSE");
return 0;
}
好了,看起来是不是很简洁明了,所以说代码复用的思想很重要!!!