泛型栈-C语言的简单实现

头文件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,&copy);
    }
    char *name = (char*)malloc(10);
    StackPop(&s, &name);
    printf("%s\n",name);
    StackDispose(&s);
    return 0;
}

用C语言实现泛型,比较麻烦的是,需要自己去计算地址。并且很容易编译通过,但是运行时候会出错。例如

void StrFree(void *elem) {
    free((char*)elem);
}
这样编译不会出现任何问题,也可以运行,但结果肯定不是你想要的。





评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值