C實現的泛型棧

typedef struct{
void *elems;    //棧底指針
int loglen;       //已經使用的長度
int alloclen;    //已經申請的長度
int elemSize;   //單個元素的長度
}stack;


void StackNew(stack *s, int elemSize)
{
assert(elemSize > 0);
s->elemSize = elemSize;
s->loglen = 0;
s->alloclen = 4;
s->elems = malloc(4 *elemSize);
assert(s->elems != NULL);
}

void StackPush(stack *s, void *elemAdd)
{
        //棧滿則採用加倍策略增長
if (s->loglen == s->alloclen)
{
          StackGrow(s);
}

void *target = (char *)(s->elems) + loglen * (s->elemSize);
memcpy(target, elemAdd, s->elemSize);
s->loglen++;
}

static  void StackGrow(stack *s)
{
     s->alloclen *= 2;
      s->elems = realloc(s->elems, (s->alloclen) * (s->elemSize));
}

void StackPop(stack *s,  void *elempop)
{
          //獲取棧頂元素指針
void *source = (char*)(s->elems) + (s->loglen - 1 ) *(s->elemSize);
memcpy(elempop, source, s->elemSize);
s->loglen--;
}

簡單測試
#include <iostream>
#include <string>
using namespace std;

class Student
{
        public:
            int id;
            string name;

            Student(){}

            Student(int i,char* n)
            {
                id = i;
                name = n;
            }
};

int main()
{
        stack s;
        StackNew(&s,sizeof(Student));

        Student *stu1 = new Student(1,"張三");

        StackPush(&s,stu1);
    
        Student stu2;
        StackPop(&s,&stu2);
    
        cout<<stu2.name<<endl;

        delete stu1;
        stu1 = 0;

        return 0;
}
發佈了4 篇原創文章 · 獲贊 9 · 訪問量 8萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章