当函数返回其值时释放malloc

void* malloc(size_t size);

void free(void* ptr);

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

int* createArray(int size) {
    int* arr = (int*)malloc(size * sizeof(int));
    if (arr == NULL) {
        printf("Failed to allocate memory.\n");
        return NULL;
    }
    for (int i = 0; i < size; i++) {
        arr[i] = i;
    }
    return arr;
}

void printArray(int* arr, int size) {
    for (int i = 0; i < size; i++) {
        printf("%d ", arr[i]);
    }
    printf("\n");
}

void destroyArray(int* arr) {
    free(arr);
}

int main() {
    int* arr = createArray(5);
    if (arr != NULL) {
        printArray(arr, 5);
        destroyArray(arr);
    }
    return 0;
}