c语言怎么定义vector数组

在C语言中，我们可以使用动态内存分配来定义一个类似于vector的数组。

首先，我们需要定义一个结构体来表示这个数组，其中包含一个指向实际数据的指针和当前数组的长度和容量。

typedef struct {
    int* data;
    int size;
    int capacity;
} Vector;

接下来，我们可以定义一些函数来对这个数组进行操作。

初始化函数：用于初始化一个空的vector数组。

void initVector(Vector* vec) {
    vec->data = NULL;
    vec->size = 0;
    vec->capacity = 0;
}

添加元素函数：用于向数组中添加一个元素，并根据需要进行内存扩容。

void addElement(Vector* vec, int element) {
    if (vec->size >= vec->capacity) {
        int new_capacity = vec->capacity * 2 + 1;
        int* new_data = (int*)malloc(new_capacity * sizeof(int));
        if (new_data == NULL) {
            // 内存分配失败的处理
            return;
        }
        memcpy(new_data, vec->data, vec->size * sizeof(int));
        free(vec->data);
        vec->data = new_data;
        vec->capacity = new_capacity;
    }
    vec->data[vec->size] = element;
    vec->size++;
}

获取元素函数：用于获取数组中指定位置的元素。

int getElement(Vector* vec, int index) {
    if (index >= 0 && index < vec->size) {
        return vec->data[index];
    } else {
        // 处理越界错误
        return 0;
    }
}

删除元素函数：用于删除数组中指定位置的元素，并根据需要进行内存收缩。

void deleteElement(Vector* vec, int index) {
    if (index >= 0 && index < vec->size) {
        for (int i = index; i < vec->size - 1; i++) {
            vec->data[i] = vec->data[i + 1];
        }
        vec->size--;
        if (vec->capacity > 2 * vec->size + 1) {
            int new_capacity = (vec->size + 1) * 2 - 1;
            int* new_data = (int*)malloc(new_capacity * sizeof(int));
            if (new_data == NULL) {
                // 内存分配失败的处理
                return;
            }
            memcpy(new_data, vec->data, vec->size * sizeof(int));
            free(vec->data);
            vec->data = new_data;
            vec->capacity = new_capacity;
        }
    } else {
        // 处理越界错误
        return;
    }
}

使用以上定义和函数，我们就可以像下面这样定义和操作一个vector数组：

int main() {
    Vector vec;
    initVector(&vec);
    
    addElement(&vec, 1);
    addElement(&vec, 2);
    addElement(&vec, 3);
    
    printf("%d\n", getElement(&vec, 0)); // 输出：1
    printf("%d\n", getElement(&vec, 1)); // 输出：2
    printf("%d\n", getElement(&vec, 2)); // 输出：3
    
    deleteElement(&vec, 1);
    
    printf("%d\n", getElement(&vec, 0)); // 输出：1
    printf("%d\n", getElement(&vec, 1)); // 输出：3
    
    return 0;
}

需要注意的是，以上代码只是简单示例，实际使用时还需要进行错误处理和内存释放等操作。

最新问答

相关标签