头文件stack.h
#ifndef STACK_H_INCLUDED
#define STACK_H_INCLUDED
typedef struct {
int elemSize; //记录所存储的类型的内存大小
int pos; //目前栈顶指针所处的位置
int allocLength; //已经分配的空间
void *elems;
void (*freefn)(void*); //释放空间的函数,当所存的数据为动态分配的时候需要用到
}stack;
void StackNew(stack *s, int elemSize, void (*freefn)(void*));
void StackDispose(stack *s);
void StackPush(stack *s, void *value);
void StackPop(stack *s, void *elemAddr);
#endif // STACK_H_INCLUDED
栈的实现stack.cpp
#include <iostream>
#include <cstring>
#include "stack.h"
void StackNew(stack *s, int elemSize,void (*freefn)(void*)) { //最后一个参数可以为NULL,表示不需要自定义的释放内存的函数
assert(elemSize > 0);
s->elems = malloc(4 * elemSize); //默认分配四个元素空间。
s->freefn = freefn;
assert(s->elems != NULL);
s->elemSize = elemSize;
s->pos = 0;
s->allocLength = 4; //四个元素空间。
}
void StackDispose(stack *s) {
if (s->freefn != NULL) {
for (int i=0; i<s->pos; i++) {
s->freefn((char*)s->elems + i * s->elemSize);
}
}
free(s->elems);
s->allocLength = 0;
s->pos = 0;
}
static void StackGrow(stack *s) { //当空间不足是重新分配更大的空间。
assert(s != NULL);
s->allocLength *= 2;
s->elems = realloc(s->elems, s->allocLength);
}
void StackPush(stack *s, void *value) {
if (s->pos >= s->allocLength)
StackGrow(s); //空间不足,增大空间
void *elemAddr = (char*)s->elems + s->pos * s->elemSize; //计算栈顶指针的位置
memcpy(elemAddr, value, s->elemSize);
s->pos++;
}
void StackPop(stack *s, void *elemAddr) {
assert(elemAddr != NULL && s != NULL);
s->pos--;
void *tem = (char*)s->elems + s->pos * s->elemSize; //取出栈顶的元素的地址
memcpy(elemAddr, tem, s->elemSize);
}
测试:
#include <stdio.h>
#include "stack.h"
void StrFree(void *elem) {
free(*(char**)elem);
}
int main()
{
char *base[] = {"ab","cd","ef"};
stack s;
StackNew(&s,sizeof(char*),StrFree);
for (int i = 0; i < 3; i++) {
char *copy = strdup(base[i]);
StackPush(&s,©);
}
char *name = (char*)malloc(10);
StackPop(&s, &name);
printf("%s\n",name);
StackDispose(&s);
return 0;
}
用C语言实现泛型,比较麻烦的是,需要自己去计算地址。并且很容易编译通过,但是运行时候会出错。例如
void StrFree(void *elem) {
free((char*)elem);
}
这样编译不会出现任何问题,也可以运行,但结果肯定不是你想要的。