c_vector

摘自：https://www.jb51.net/program/311452tbw.htm

 

// 创建一个空的Vector
MY_VECTOR* vector_create() {
    MY_VECTOR *v = (MY_VECTOR*)calloc(1, sizeof(MY_VECTOR));
    if (v == NULL) {
        puts("error：创建一个空的Vector时分配内存失败");
        exit(-1);
    }
    // 给Vector的成员变量赋值
    v->data = c1alloc(DEFAULT_CAPACITY, sizeof(Element));
    if (v->data == NULL) {
        puts("error：创建一个空的Vector时分配内存失败");
        free(v); // 因为下面一行是直接退出程序，free(v)意义不大，但最好还是写上，养成习惯
        exit(-1);
    }
    v->capacity = DEFAULT_CAPACITY;
    return v;
}

// 销毁释放Vector
void vector_destroy(MY_VECTOR *v) {
    if (v == NULL) { // Vector不能是NULL
        return;
    }
    free(v->data);
    free(v);
}

// 向动态数组的末尾新增一个元素
void vector_push_back(MY_VECTOR *v, Element val) {
    if (v == NULL) { // Vector不能是NULL
        return;
    }
    if (v->size == v->capacity) { // 容量不足，扩容
        vector_rsize(v);
    }
    v->data[v->size] = val;
    v->size++;
}

// 向数组的前面插入一个元素
void vector_push_front(MY_VECTOR *v, Element val) {
    if (v == NULL) { // Vector不能是NULL
        return;
    }
    if (v->size == v->capacity) { // 容量不足，扩容
        vector_rsize(v);
    }
    // 从下标0向后移动并在0下标位置赋值
    move_data(v, 0);
    v->data[0] = val;
    v->size++;
}

// 将元素val添加到索引为idx的位置，idx后面的元素依次后移
void vector_insert(MY_VECTOR *v, int idx, Element val) {
    if (v == NULL || idx < 0 || idx > v->size) { // Vector不能是NULL，索引位置不能为负且不能越界
        return;
    }
    if (v->size == v->capacity) { // 容量不足，扩容
        vector_rsize(v);
    }
    move_data(v, idx);
    v->data[idx] = val;
    v->size++;
}

/*
给Vector的动态数组扩容

tips：此函数以下几点需要注意

算术运算‘+’的优先级比位运算符‘>>’高，要用括号括起来；
用realloc扩容，不能用calloc和malloc来扩大容量，数据会丢失；
扩容的时候要注意是给Vector的data数组扩容，即v->data；
只有calloc会默认自动赋初值，malloc和realloc都不会默认赋初值，记得给扩容部分附上初始值；
一定要先执行v->data=temp，之后再初始化新开辟部分；因为空间不足的情况下，realloc不一定在原内存地址后面扩容！！
*/
static void vector_rsize(MY_VECTOR *v) {
    int old_capacity = v->capacity;
    // tips：算术运算‘+’的优先级比位运算符‘>>’高，要用括号括起来
    int new_capacity = v->size < HIGHT_SIZE ? old_capacity << 1 : old_capacity + (old_capacity >> 1);
    // tips：用realloc扩容，不能用calloc和malloc，数据会丢失！
    // tips：扩容的是v->data，而不是v
    Element *temp = realloc(v->data, new_capacity * sizeof(Element));
    if (temp == NULL) {
        puts("error：给Vector的动态数组扩容失败");
        exit(-1);
    }
    // 更新Vector信息
    v->data = temp;
    v->capacity = new_capacity;
    // tips：v->data扩容部分附上初值，且此操作要在v->data=temp之后
    memset(v->data + v->size, 0, (v->capacity - v->size) * sizeof(Element));
}

// 将数组的元素从指定下标 idx 位置依次向后挪动1个位置
static void vector_move_data(MY_VECTOR *v, int idx) {
    for (int i = v->size - 1; i >= idx; i--) {
        v->data[i+1] = v->data[i];
    }
    v->data[idx] = 0;
}

// 将数组的index位置元素取出, 下标位置后面的依次向前挪动
static void vector_remove_data(MY_VECTOR *v, int idx) {
    for (int i = v->size - 1; i >= idx; i--) {
        v->data[i+1] = v->data[i];
    }
    v->data[idx] = 0;
}

#ifndef __MY_VECTOR_H__
#define __MY_VECTOR_H__

// 数组默认容量设置为10
#define DEFAULT_CAPACITY 10 
// 数组长度低于 HIGHT_SIZE 时，每次按照原长度翻倍扩容
// 数组长度高于 HIGHT_SIZE 时，扩充原容量的1/2
#define HIGHT_SIZE 1024 

// 定义存储的数据类型
typedef void* Element;

// 定义Vector的数据结构
typedef struct _my_vector {
    Element *data; // 用于存储数据的动态数组的指针
    int size; // 数组长度
    int capacity; // 容量
}MY_VECTOR;

// 创建一个空的Vector
MY_VECTOR* vector_create();

// 销毁释放Vector
void vector_destroy(MY_VECTOR *v);

// 向动态数组的末尾新增一个元素
void vector_push_back(MY_VECTOR *v, Element val);

// 向数组的前面插入一个元素
void vector_push_front(MY_VECTOR *v, Element val);

// 将元素val添加到索引为idx的位置，index后面的元素依次后移
void vector_insert(MY_VECTOR *v, int index, Element val);

// 给Vector的动态数组扩容
static void vector_rsize(MY_VECTOR *v);

// 将数组的元素从指定下标位置依次向后挪动
static void vector_move_data(MY_VECTOR *v, int index);

// 将数组的index位置元素取出, 下标位置后面的依次向前挪动
static Element vector_remove_data(MY_VECTOR *v, int index);


#endif // end for __MY_VECTOR_H__