C动态数组_添加元素后动态扩容

/*
C动态数组，可以在容量不足，添加元素后动态扩容
*/

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

#define INITIAL_CAPACITY 5

// 定义一个结构体，封装动态数组及其相关信息
typedef struct {
    int* data;
    int capacity;
    int count;
} AutoArray;

// 初始化动态数组
AutoArray initArray() {
    AutoArray arr = {NULL, 0, 0};
    arr.data = (int*) malloc(INITIAL_CAPACITY * sizeof(int));
    if (arr.data == NULL) {
        fprintf(stderr, "Memory allocation failed.\n");
        exit(EXIT_FAILURE);
    }
    arr.capacity = INITIAL_CAPACITY;
    return arr;
}

// 向动态数组添加一个元素，自动扩容
void addElement(AutoArray* arr, int value) {
    if (arr->count >= arr->capacity) {
        // 扩容操作：增加一倍容量
        int newCapacity = arr->capacity * 2;
        int* newData = (int*) realloc(arr->data, newCapacity * sizeof(int));
        if (newData == NULL) {
            fprintf(stderr, "Memory reallocation failed.\n");
            exit(EXIT_FAILURE);
        }
        arr->data = newData;
        arr->capacity = newCapacity;
    }
    // 添加新元素到数组末尾
    arr->data[arr->count++] = value;
}

// 释放动态数组占用的内存
void freeArray(AutoArray* arr) {
    free(arr->data);
    arr->data = NULL;
    arr->capacity = arr->count = 0;
}

int main() {
    AutoArray arr = initArray();
    // 向动态数组添加多个元素
    addElement(&arr, 1);
    addElement(&arr, 2);
    addElement(&arr, 3);
    addElement(&arr, 4);
    addElement(&arr, 5); // 初始容量已满，触发扩容
    printf("arr.capacity:%d\n",arr.capacity);
    addElement(&arr, 6);
    printf("arr.capacity:%d\n",arr.capacity);
    addElement(&arr, 7);
    addElement(&arr, 8);
    addElement(&arr, 9);
    addElement(&arr, 10);
    addElement(&arr, 11);
    printf("arr.capacity:%d\n",arr.capacity);
    addElement(&arr, 12);
    addElement(&arr, 13);
    addElement(&arr, 14);
    
	int i;
    // 打印数组内容
    for (i = 0; i < arr.count; ++i) {
        printf("arr.data[%d]:%d\n",i,arr.data[i]);
    }
    // 不再需要数组时，释放内存
    freeArray(&arr);
    return 0;
}