教你从零开始写一个哈希表--哈希表结构

  我们的键-值对（items），每一个都会被保存在结构体中：

// hash_table.h
typedef struct {
    char* key;
    char* value;
} ht_item;

  我们的哈希表保存了一组键值对的指针，以及哈希表大小的一些细节和它的填满情况：

// hash_table.h
typedef struct {
    int size;
    int count;
    ht_item** items;
} ht_hash_table;

初始化和删除

  我们需要为ht_item定义初始化幻术。这个函数分配了ht_item大小的内存，同时在新的内存块中保存了字符串k和v的副本。把这个函数标记为static，是因为它只会被哈希表内部调用一次。

// hash_table.c
#include <stdlib.h>
#include <string.h>

#include "hash_table.h"

static ht_item* ht_new_item(const char* k, const char* v) {
    ht_item* i = malloc(sizeof(ht_item));
    i->key = strdup(k);
    i->value = strdup(v);
    return i;
}

  ht_new函数初始化了一个新的哈希表。size变量定义了我们可以保存多少个键值对。此处写死了53个。我们会在“重置大小”一节进行扩展。我们用calloc函数初始化一组键值对。calloc这个函数用NULL对分配的内存进行填充。数组的首位存储是NULL意味着这个桶是空的。

// hash_table.c
ht_hash_table* ht_new() {
    ht_hash_table* ht = malloc(sizeof(ht_hash_table));

    ht->size = 53;
    ht->count = 0;
    ht->items = calloc((size_t)ht->size, sizeof(ht_item*));
    return ht;
}

  我们也需要函数来删除ht_items变量和ht_hash+tables结构。这将会释放我们已经分配的内存，不然就会导致内存泄漏。

// hash_table.c
static void ht_del_item(ht_item* i) {
    free(i->key);
    free(i->value);
    free(i);
}


void ht_del_hash_table(ht_hash_table* ht) {
    for (int i = 0; i < ht->size; i++) {
        ht_item* item = ht->items[i];
        if (item != NULL) {
            ht_del_item(item);
        }
    }
    free(ht->items);
    free(ht);
}

  我们已经写完了定义哈希表的代码。它让我们可以创建和销毁一个哈希表。尽管它这个时候还做不了什么事情，但我们依然可以试运行一下这份代码。

// main.c
#include "hash_table.h"

int main() {
    ht_hash_table* ht = ht_new();
    ht_del_hash_table(ht);
}

上一篇:教你从零开始写一个哈希表–导读
下一篇:教你从零开始写一个哈希表–哈希函数