编写自己的malloc

最新推荐文章于 2023-08-22 08:00:00 发布

最新推荐文章于 2023-08-22 08:00:00 发布 · 281 阅读

算法专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种自定义内存分配器的实现方法，通过减少对标准库函数malloc的依赖来提高程序效率，并提供了一个使用该内存分配器进行单词计数的实际应用示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1,如果一个程序大量调用malloc,程序的很多时间将会消耗在malloc函数的调用上.
2,编写自己的malloc.
给出一段模板代码:


#define NODESIZE 8
#define NODEGROUP 100
int nodeleft=0;
char* freenode;

void* myMalloc(int size)
{
    void* p;
    if(size!=NODESIZE)
        return malloc(size);
    if(nodeleft==0)
    {
        freenode=malloc(NODESIZE*NODEGROUP);
        nodeleft=NODEGROUP;
    }
    nodeleft--;
    p=(void*)freenode;
    freenode+=NODESIZE;
    return p;
}

3,统计单词个数的那个实例:


#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct node *nodeptr;
typedef struct node
{
	char *word;
	int count;
	nodeptr next;
} node;

#define NHASH 29989
#define MULT 31
nodeptr bin[NHASH];

unsigned int hash(char *p)
{
    unsigned int h = 0;
	for ( ; *p; p++)
		h = MULT * h + *p;
	return h % NHASH;
}

#define NODEGROUP 1000
int nodesleft = 0;

//注意这里freenode的声明
nodeptr freenode;
nodeptr nmalloc()
{
    if (nodesleft == 0)
    {
		freenode = malloc(NODEGROUP*sizeof(node));
		nodesleft = NODEGROUP;
	}
	nodesleft--;
	return freenode++;
}

#define CHARGROUP 10000
int charsleft = 0;
char *freechar;

char *smalloc(int n)
{
    if (charsleft < n) {
		freechar = malloc(n+CHARGROUP);
		charsleft = n+CHARGROUP;
	}
	charsleft -= n;
	freechar += n; //移动freechar
	return freechar - n;
}

void incword(char *s)
{
    nodeptr p;
	int h = hash(s);
	for (p = bin[h]; p != NULL; p = p->next)
		if (strcmp(s, p->word) == 0) {
			(p->count)++;
			return;
		}
	p = (nodeptr)nmalloc();
	p->count = 1;
	p->word = smalloc(strlen(s)+1);
	strcpy(p->word, s);
	p->next = bin[h];
	bin[h] = p;
}

int main()
{
    freopen("genetic.txt","r",stdin);
    int i;
	nodeptr p;
	char buf[100];
	for (i = 0; i < NHASH; i++)
		bin[i] = NULL;
	while (scanf("%s", buf) != EOF)
		incword(buf);
	for (i = 0; i < NHASH; i++)
		for (p = bin[i]; p != NULL; p = p->next)
			printf("%s %d\n", p->word, p->count);
	return 0;
}