释放 C 中递归子例程中的内存

Question

我想问一个关于释放C语言内存的问题。我正在实现合并排序函数，如下所示：

合并子例程：

int* merge (int* array_left, unsigned int left_length, int* array_right, unsigned int right_length) {
    unsigned int result_size = right_length + left_length;
    int* result = malloc(result_size*sizeof(int));
    int r = 0; // result index 
    // Iterate through all left and right array elements
    int i = 0;  // left index
    int j = 0;  // right index
    while ( (i < left_length) && (j < right_length) ) {
        if ( *(array_left+i) < *(array_right+j) ) {
            *(result+r) = *(array_left+i);
            i++;
        } else {
            *(result+r) = *(array_right+j);
            j++;
        }
        r++;
    }
    // Fill the remaining elements to the result
    if (i < left_length)
        while (i < left_length) {
            *(result+r) = *(array_left+i);
            r++;
            i++;
        }
    if (j < right_length)
        while (j < right_length) {
            *(result+r) = *(array_right+j);
            r++;
            j++;
        }
    return result;
}

合并排序：

   int* mergeSort(int* array, unsigned int length) {
      // Base case
    if (length <= 1)
        return array;
    // Middle element
    unsigned int middle = length / 2;
    int* array_right =  mergeSort(array, middle);
    int* array_left = mergeSort(&array[middle], length-middle);
    // Result is merge from two shorted right and left array
    int* result = merge(array_left, length-middle, array_right, middle);
    return result;
}

程序运行正常，但我没有从我的 malloc 调用中释放内存，实际上我无法弄清楚如何放置 free（）。我试图释放array_right和array_left但我收到错误，告诉我我只能释放 malloc 直接分配的指针。

请帮忙！提前谢谢你们。

Answer 1

你需要添加

free(arrayLeft);
free(arrayRight);

并且还 malloc 并复制数组，即使它的长度在 mergeSort 中为 1：

int* mergeSort(int* array, unsigned int length) {
    // Base case
    if (!length) return NULL;
    if (length == 1) {
        // Make a copy of a single-element array
        int *tmp = malloc(sizeof(int));
        *tmp = *array;
        return tmp;
    }
    ... // The rest of your code
}

这将确保 mergeSort 的调用方始终拥有它返回的数组，因此他必须在所有情况下释放它。

当您尝试它时它不起作用的原因是您没有复制琐碎的数组，这导致双重释放其中一些数组。