在C中递归子程序中释放内存

时间:2013-09-07 02:32:41

标签: c malloc free mergesort

我想问一个关于在C中释放内存的问题。我正在实现mergeSort函数如下:

合并子程序:

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直接分配的指针。

请帮忙!提前谢谢你们。

1 个答案:

答案 0 :(得分:3)

您需要添加

free(arrayLeft);
free(arrayRight);

并且即使在mergeSort中长度为1的情况下也可以复制malloc并复制数组:

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 的调用者始终拥有它返回的数组,因此他必须在所有情况下释放它。

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